xinyun/ComfyUI-CogVideoXWrapper
1
0
Fork 0
mirror of https://git.datalinker.icu/kijai/ComfyUI-CogVideoXWrapper.git synced 2025-12-10 21:44:28 +08:00
Code Issues Packages Projects Releases Wiki Activity

Update model_loading.py

Browse Source
This commit is contained in:
kijai 2024-11-12 00:22:26 +02:00
parent 00fde5ebce
commit db697fea11
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
model_loading.py
View File

@ -113,6 +113,9 @@ class DownloadAndLoadCogVideoModel:
def loadmodel(self, model, precision, fp8_transformer="disabled", compile="disabled", def loadmodel(self, model, precision, fp8_transformer="disabled", compile="disabled",
enable_sequential_cpu_offload=False, pab_config=None, block_edit=None, lora=None, compile_args=None, enable_sequential_cpu_offload=False, pab_config=None, block_edit=None, lora=None, compile_args=None,
attention_mode="sdpa", load_device="main_device"): attention_mode="sdpa", load_device="main_device"):
if precision == "fp16" and "1.5" in model:
raise ValueError("1.5 models do not work in fp16")
device = mm.get_torch_device() device = mm.get_torch_device()
offload_device = mm.unet_offload_device() offload_device = mm.unet_offload_device()