aaa

app.py

@@ -58,7 +58,7 @@ example_inputs = [
     "examples/9f2909760aff526070f169620ff38290.mp4",
     # "examples/18.mp4",
     # "examples/27.mp4",
-    # "examples/28.mp4",
+    "examples/28.mp4",
     "examples/73fc0b2a3af3474de27c7da0bfbf5faa.mp4",
     # "examples/episode_48-camera_third_view.mp4",
     "examples/extra_5.mp4",

dkt/pipelines/pipeline.py

@@ -222,10 +222,10 @@ class ModelConfig:
                     # else:
                     hf_snap_download(
                         repo_id=self.model_id,
-                        revision = 'abd742fbb0d2a2f5e231e5ca3e9dd61e264a87c2',
+                        revision = 'main',
                         local_dir=os.path.join(local_model_path, self.model_id),
                         allow_patterns=allow_file_pattern,
-                        # ignore_patterns=downloaded_files if downloaded_files else None
+                        ignore_patterns=downloaded_files if downloaded_files else None
                     )
 
             # Let rank 1, 2, ... wait for rank 0
@@ -872,8 +872,8 @@ class DKTPipeline:
         
         lora_config = ModelConfig(
             model_id=model_id ,
-            # origin_file_pattern="*.safetensors",
-            origin_file_pattern="TransPhy3D_cleargrasp_HISS_DREDS_1.3B_depth_70K_lora.safetensors",
+            origin_file_pattern="*.safetensors",
+            # origin_file_pattern="TransPhy3D_cleargrasp_HISS_DREDS_1.3B_depth_70K_lora.safetensors",
             offload_device="cpu",
         )
         lora_config.download_if_necessary(use_usp=False)