Spaces:

roll-ai
/

Sci-Fi

Paused

AhmadMustafa commited on Oct 3

Commit

1a2f974

1 Parent(s): 6cd58a3

rename

Files changed (1) hide show

app.py CHANGED Viewed

@@ -46,7 +46,7 @@ def load_pipeline(dtype_str="bfloat16"):
     # Set paths
     model_base_path = repo_path
     cogvideo_path = os.path.join(model_base_path, "CogVideoX-5b-I2V")
-    ef_net_path = os.path.join(model_base_path, "EF_Net", "EF_Net.pth")
     print(f"CogVideo path: {cogvideo_path}")
     print(f"EF-Net path: {ef_net_path}")

     # Set paths
     model_base_path = repo_path
     cogvideo_path = os.path.join(model_base_path, "CogVideoX-5b-I2V")
+    ef_net_path = os.path.join(model_base_path, "EF_Net", "EF_Net.pt")
     print(f"CogVideo path: {cogvideo_path}")
     print(f"EF-Net path: {ef_net_path}")