xinyun/ComfyUI-CogVideoXWrapper
1
0
Fork 0
mirror of https://git.datalinker.icu/kijai/ComfyUI-CogVideoXWrapper.git synced 2025-12-15 07:54:24 +08:00
Code Issues Packages Projects Releases Wiki Activity

peft

Browse Source
This commit is contained in:
kijai 2024-10-06 11:53:38 +03:00
parent d8a9f31de9
commit fc7e7d7c62
2 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
nodes.py
View File

@ -58,7 +58,7 @@ from .cogvideox_fun.autoencoder_magvit import AutoencoderKLCogVideoX as Autoenco
from .cogvideox_fun.utils import get_image_to_video_latent, get_video_to_video_latent, ASPECT_RATIO_512, get_closest_ratio, to_pil from .cogvideox_fun.utils import get_image_to_video_latent, get_video_to_video_latent, ASPECT_RATIO_512, get_closest_ratio, to_pil
from .cogvideox_fun.pipeline_cogvideox_inpaint import CogVideoX_Fun_Pipeline_Inpaint from .cogvideox_fun.pipeline_cogvideox_inpaint import CogVideoX_Fun_Pipeline_Inpaint
from .cogvideox_fun.pipeline_cogvideox_control import CogVideoX_Fun_Pipeline_Control from .cogvideox_fun.pipeline_cogvideox_control import CogVideoX_Fun_Pipeline_Control
from .cogvideox_fun.lora_utils import merge_lora, load_lora_into_transformer
from PIL import Image from PIL import Image
import numpy as np import numpy as np
import json import json
@ -346,6 +346,7 @@ class DownloadAndLoadCogVideoModel:
transformer = transformer.to(dtype).to(offload_device) transformer = transformer.to(dtype).to(offload_device)
if lora is not None: if lora is not None:
from .cogvideox_fun.lora_utils import merge_lora, load_lora_into_transformer
logging.info(f"Merging LoRA weights from {lora['path']} with strength {lora['strength']}") logging.info(f"Merging LoRA weights from {lora['path']} with strength {lora['strength']}")
if "fun" in model.lower(): if "fun" in model.lower():
transformer = merge_lora(transformer, lora["path"], lora["strength"]) transformer = merge_lora(transformer, lora["path"], lora["strength"])

3
requirements.txt
View File

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