fix flux

Artheriax · Sep 4, 2024 · ce94b5a · ce94b5a
1 parent f2b84b1
commit ce94b5a
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/modules/model_flux.py b/modules/model_flux.py
@@ -46,7 +46,7 @@ def load_flux_quanto(checkpoint_info, diffusers_load_config):
         quantization_map = os.path.join(repo_path, "transformer", "quantization_map.json")
         debug(f'Loading FLUX: quantization map="{quantization_map}" repo="{checkpoint_info.name}" component="transformer"')
         if not os.path.exists(quantization_map):
-            repo_id = checkpoint_info.name.replace('Diffusers/', '').replace('models--', '').replace('--', '/')
+            repo_id = checkpoint_info.name.replace('Diffusers/', '').replace('Diffusers\\', '').replace('models--', '').replace('--', '/')
             quantization_map = hf_hub_download(repo_id, subfolder='transformer', filename='quantization_map.json', **diffusers_load_config)
         with open(quantization_map, "r", encoding='utf8') as f:
             quantization_map = json.load(f)
@@ -71,7 +71,7 @@ def load_flux_quanto(checkpoint_info, diffusers_load_config):
         quantization_map = os.path.join(repo_path, "text_encoder_2", "quantization_map.json")
         debug(f'Loading FLUX: quantization map="{quantization_map}" repo="{checkpoint_info.name}" component="text_encoder_2"')
         if not os.path.exists(quantization_map):
-            repo_id = checkpoint_info.name.replace('Diffusers/', '').replace('models--', '').replace('--', '/')
+            repo_id = checkpoint_info.name.replace('Diffusers/', '').replace('Diffusers\\', '').replace('models--', '').replace('--', '/')
             quantization_map = hf_hub_download(repo_id, subfolder='text_encoder_2', filename='quantization_map.json', **diffusers_load_config)
         with open(quantization_map, "r", encoding='utf8') as f:
             quantization_map = json.load(f)
@@ -153,7 +153,7 @@ def load_transformer(file_path): # triggered by opts.sd_unet change
 
 def load_flux(checkpoint_info, diffusers_load_config): # triggered by opts.sd_checkpoint change
     quant = get_quant(checkpoint_info.path)
-    repo_id = checkpoint_info.name.replace('Diffusers/', '').replace('models--', '').replace('--', '/')
+    repo_id = checkpoint_info.name.replace('Diffusers/', '').replace('Diffusers\\', '').replace('models--', '').replace('--', '/')
     shared.log.debug(f'Loading FLUX: model="{checkpoint_info.name}" repo="{repo_id}" unet="{shared.opts.sd_unet}" t5="{shared.opts.sd_text_encoder}" vae="{shared.opts.sd_vae}" quant={quant} offload={shared.opts.diffusers_offload_mode} dtype={devices.dtype}')
     debug(f'Loading FLUX: config={diffusers_load_config}')
     modelloader.hf_login()