nodes/aigc-apps-VideoX-Fun
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Rename && Update Readme

Browse Source
This commit is contained in:
bubbliiiing
2024-09-10 11:22:35 +08:00
parent d29d193177
commit ef55bd3bbe
15 changed files with 107 additions and 132 deletions
Download Patch File Download Diff File Expand all files Collapse all files
predict_i2v.py
+4 -4
View File
@@ -15,8 +15,8 @@ from PIL import Image
from cogvideox.models.transformer3d import CogVideoXTransformer3DModel
from cogvideox.models.autoencoder_magvit import AutoencoderKLCogVideoX
from cogvideox.pipeline.pipeline_cogvideox import CogVideoX_FUN_Pipeline
from cogvideox.pipeline.pipeline_cogvideox_inpaint import CogVideoX_FUN_Pipeline_Inpaint
from cogvideox.pipeline.pipeline_cogvideox import CogVideoX_Fun_Pipeline
from cogvideox.pipeline.pipeline_cogvideox_inpaint import CogVideoX_Fun_Pipeline_Inpaint
from cogvideox.utils.lora_utils import merge_lora, unmerge_lora
from cogvideox.utils.utils import get_image_to_video_latent, save_videos_grid
@@ -112,7 +112,7 @@ scheduler = Choosen_Scheduler.from_pretrained(
)
if transformer.config.in_channels != vae.config.latent_channels:
pipeline = CogVideoX_FUN_Pipeline_Inpaint.from_pretrained(
pipeline = CogVideoX_Fun_Pipeline_Inpaint.from_pretrained(
model_name,
vae=vae,
text_encoder=text_encoder,
@@ -121,7 +121,7 @@ if transformer.config.in_channels != vae.config.latent_channels:
torch_dtype=weight_dtype
)
else:
pipeline = CogVideoX_FUN_Pipeline.from_pretrained(
pipeline = CogVideoX_Fun_Pipeline.from_pretrained(
model_name,
vae=vae,
text_encoder=text_encoder,