Added support for CausVid Lora and MoviiGen

This commit is contained in:
DeepBeepMeep
2025-05-20 20:03:30 +02:00
parent 3e635f32b3
commit a9119782d0
8 changed files with 67 additions and 158 deletions

View File

@@ -105,12 +105,6 @@ def load_i2v_model(model_filename, text_encoder_filename, is_720p):
wan_model = wan.WanI2V(
config=cfg,
checkpoint_dir=DATA_DIR,
device_id=0,
rank=0,
t5_fsdp=False,
dit_fsdp=False,
use_usp=False,
i2v720p=True,
model_filename=model_filename,
text_encoder_filename=text_encoder_filename
)
@@ -120,12 +114,6 @@ def load_i2v_model(model_filename, text_encoder_filename, is_720p):
wan_model = wan.WanI2V(
config=cfg,
checkpoint_dir=DATA_DIR,
device_id=0,
rank=0,
t5_fsdp=False,
dit_fsdp=False,
use_usp=False,
i2v720p=False,
model_filename=model_filename,
text_encoder_filename=text_encoder_filename
)
@@ -624,8 +612,8 @@ def main():
# Actually run the i2v generation
try:
sample_frames = wan_model.generate(
user_prompt,
input_img,
input_prompt = user_prompt,
image_start = input_img,
frame_num=frame_count,
width=width,
height=height,