jinx/ComfyUI
1
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2025-08-02 15:04:50 +08:00
Code Issues Activity

Add model_options to load_controlnet function.

Browse Source
This commit is contained in:
comfyanonymous
2024-09-19 05:01:00 -04:00
parent de8e8e3b0d
commit ad66f7c7d8
2 changed files with 32 additions and 26 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
comfy/sd.py
View File

@@ -645,7 +645,7 @@ def load_diffusion_model_state_dict(sd, model_options={}): #load unet in diffuse
manual_cast_dtype = model_management.unet_manual_cast(unet_dtype, load_device, model_config.supported_inference_dtypes)
model_config.set_inference_dtype(unet_dtype, manual_cast_dtype)
model_config.custom_operations = model_options.get("custom_operations", None)
model_config.custom_operations = model_options.get("custom_operations", model_config.custom_operations)
model = model_config.get_model(new_sd, "")
model = model.to(offload_device)
model.load_model_weights(new_sd, "")