diff --git a/modules/model_flux.py b/modules/model_flux.py index 4e2bf14eb..01b093712 100644 --- 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()