Update app.py

app.py

@@ -29,16 +29,14 @@ class PatchedModelManager(BaseModelManager):
     def __init__(self, *args, **kwargs):
         super().__init__(*args, **kwargs)
 
-        # Your custom architecture dict entries to patch or add
+        # Patch architecture dict here
         custom_architecture_dict = {
             "WanModel": ("diffsynth.models.wan_model", "WanModelPusa", None),
-            # Add more fixes if needed
         }
-
-        # Update or replace the architecture_dict
         self.architecture_dict.update(custom_architecture_dict)
 
 
+
 # Constants
 WAN_SUBFOLDER = "Wan2.1-T2V-14B"
 MODEL_REPO_ID = "RaphaelLiu/PusaV1"
@@ -98,6 +96,8 @@ def generate_video(prompt: str):
     #     device="cuda"
     # )
 
+    
+
     manager = PatchedModelManager(
        file_path_list=[WAN_MODEL_PATH],
        torch_dtype=torch.float16,