rename

app.py

@@ -46,11 +46,21 @@ 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.pt")
+    ef_net_path = os.path.join(
+        model_base_path, "EF_Net", "EF_Net.pt"
+    )  # Changed from .pth to .pt
 
     print(f"CogVideo path: {cogvideo_path}")
     print(f"EF-Net path: {ef_net_path}")
 
+    # Verify the EF_Net file exists
+    if not os.path.exists(ef_net_path):
+        # Try to list files in the EF_Net directory to debug
+        ef_net_dir = os.path.join(model_base_path, "EF_Net")
+        if os.path.exists(ef_net_dir):
+            print(f"Files in EF_Net directory: {os.listdir(ef_net_dir)}")
+        raise FileNotFoundError(f"EF-Net weights not found at {ef_net_path}")
+
     # Load models
     print("Loading tokenizer and text encoder...")
     tokenizer = T5Tokenizer.from_pretrained(os.path.join(cogvideo_path, "tokenizer"))
@@ -73,9 +83,6 @@ def load_pipeline(dtype_str="bfloat16"):
 
     # Load EF-Net
     print(f"Loading EF-Net from {ef_net_path}...")
-    if not os.path.exists(ef_net_path):
-        raise FileNotFoundError(f"EF-Net weights not found at {ef_net_path}")
-
     EF_Net_model = (
         EF_Net(num_layers=4, downscale_coef=8, in_channels=2, num_attention_heads=48)
         .requires_grad_(False)