adding CogVideox1.5-5B-I2V

download/download.py

@@ -17,7 +17,7 @@ def download_model():
         local_dir_use_symlinks=False,
     )
     snapshot_download(
-        repo_id="THUDM/CogVideoX1.5-5B-SAT",
+        repo_id="THUDM/CogVideoX1.5-5B-I2V",
         local_dir="/app/pretrained/CogVideoX-5b-I2V",
         local_dir_use_symlinks=False,
     )

inference/cli_demo_camera_i2v_pcd.py

@@ -11,7 +11,8 @@ from transformers import T5EncoderModel, T5Tokenizer
 from diffusers import (
     CogVideoXDDIMScheduler,
     CogVideoXDPMScheduler,
-    AutoencoderKLCogVideoX
+    AutoencoderKLCogVideoX,
+    CogVideoXImageToVideoPipeline
 )
 from diffusers.utils import export_to_video, load_video
 from torchvision.transforms import ToPILImage