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

check

Browse Source
This commit is contained in:
kijai 2024-11-09 15:41:57 +02:00
parent 1c3aff9000
commit 9a797229f2
2 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
nodes.py
View File

@ -5,6 +5,8 @@ import comfy.model_management as mm
from einops import rearrange from einops import rearrange
from contextlib import nullcontext from contextlib import nullcontext
from .utils import log, check_diffusers_version
check_diffusers_version()
from diffusers.schedulers import ( from diffusers.schedulers import (
CogVideoXDDIMScheduler, CogVideoXDDIMScheduler,
CogVideoXDPMScheduler, CogVideoXDPMScheduler,
@ -42,7 +44,7 @@ from PIL import Image
import numpy as np import numpy as np
import json import json
from .utils import log, check_diffusers_version
script_directory = os.path.dirname(os.path.abspath(__file__)) script_directory = os.path.dirname(os.path.abspath(__file__))

2
requirements.txt
View File

@ -1,5 +1,5 @@
huggingface_hub huggingface_hub
diffusers>=0.30.3 diffusers>=0.31.0
accelerate>=0.33.0 accelerate>=0.33.0
einops einops
peft peft