xinyun/ComfyUI-CogVideoXWrapper
1
0
Fork 0
mirror of https://git.datalinker.icu/kijai/ComfyUI-CogVideoXWrapper.git synced 2026-05-21 18:07:09 +08:00
Code Issues Packages Projects Releases Wiki Activity

Update model_loading.py

Browse Source
This commit is contained in:
kijai 2024-11-08 21:23:29 +02:00
parent 4c2ce52f57
commit f7a88cbd56
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
model_loading.py
View File

@ -4,6 +4,8 @@ import torch.nn as nn
import json import json
import folder_paths import folder_paths
import comfy.model_management as mm import comfy.model_management as mm
from .utils import check_diffusers_version, remove_specific_blocks, log
check_diffusers_version()
from diffusers.models import AutoencoderKLCogVideoX from diffusers.models import AutoencoderKLCogVideoX
from diffusers.schedulers import CogVideoXDDIMScheduler from diffusers.schedulers import CogVideoXDDIMScheduler
@ -20,7 +22,6 @@ from .cogvideox_fun.pipeline_cogvideox_control import CogVideoX_Fun_Pipeline_Con
from .videosys.cogvideox_transformer_3d import CogVideoXTransformer3DModel as CogVideoXTransformer3DModelPAB from .videosys.cogvideox_transformer_3d import CogVideoXTransformer3DModel as CogVideoXTransformer3DModelPAB
from .utils import check_diffusers_version, remove_specific_blocks, log
from comfy.utils import load_torch_file from comfy.utils import load_torch_file
script_directory = os.path.dirname(os.path.abspath(__file__)) script_directory = os.path.dirname(os.path.abspath(__file__))
@ -311,7 +312,6 @@ class DownloadAndLoadCogVideoGGUFModel:
"pab_config": ("PAB_CONFIG", {"default": None}), "pab_config": ("PAB_CONFIG", {"default": None}),
"block_edit": ("TRANSFORMERBLOCKS", {"default": None}), "block_edit": ("TRANSFORMERBLOCKS", {"default": None}),
"compile": (["disabled","torch"], {"tooltip": "compile the model for faster inference, these are advanced options only available on Linux, see readme for more info"}), "compile": (["disabled","torch"], {"tooltip": "compile the model for faster inference, these are advanced options only available on Linux, see readme for more info"}),
} }
} }