298977887/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2025-04-15 16:13:29 +00:00
Code Issues Packages Projects Releases Wiki Activity

Move the model_sampling to CPU since it is very small amount of work

Browse Source
This commit is contained in:
David McCloskey 2025-01-17 12:18:30 -06:00
parent 7fc3ccdcc2
commit 6a737925b1
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
comfy/sd.py
View File

@ -923,6 +923,8 @@ def load_state_dict_guess_config(sd, output_vae=True, output_clip=True, output_c
if inital_load_device != torch.device("cpu"):
logging.info("loaded diffusion model directly to GPU")
model_management.load_models_gpu([model_patcher], force_full_load=True)
# damcclos: move the model_sampling back to the CPU. The work needed for this is not worth the gpu.
model_patcher.model.model_sampling.to(torch.device("cpu"))
return (model_patcher, clip, vae, clipvision)