Update model_loader.py

model_loader.py

@@ -308,7 +308,7 @@ def load_all_models(self, progress_callback: Optional[callable] = None, cancel_e
             
             matanyone_time = time.time() - matanyone_start
             self.loading_stats['matanyone_load_time'] = matanyone_time
-            logger.info(f"MatAnyone loaded in {matanyone_time:.2f}s")
+            logger.info(f"MatAnyone loaded in {matanyone_time:.1f}s")
             
             # Final setup
             total_time = time.time() - start_time
@@ -499,20 +499,12 @@ def _load_matanyone_model(self, progress_callback: Optional[callable] = None):
 # ============================================================================ #
 
     def _load_matanyone_strategy_1(self):
-        """MatAnyone loading strategy 1: Direct model import"""
-        from matanyone.model.matanyone import MatAnyOne
-        from matanyone.inference.inference_core import InferenceCore
-        
-        cfg = OmegaConf.create({
-            'model': {'name': 'MatAnyOne'},
-            'device': str(self.device),
-            'fp16': True if self.device.type == 'cuda' else False
-        })
-        
-        net = MatAnyOne(cfg)
-        core = InferenceCore(net, cfg)
+        """MatAnyone loading strategy 1: Official HuggingFace InferenceCore"""
+        from matanyone import InferenceCore
         
-        return net, core
+        # Initialize with the official model repo
+        processor = InferenceCore("PeiqingYang/MatAnyone")
+        return processor, processor
     
     def _load_matanyone_strategy_2(self):
         """MatAnyone loading strategy 2: Alternative import paths"""
@@ -550,20 +542,11 @@ def _load_matanyone_strategy_3(self):
         return model, engine
     
     def _load_matanyone_strategy_4(self):
-        """MatAnyone loading strategy 4: Hugging Face Hub approach"""
-        from huggingface_hub import hf_hub_download
-        from matanyone import load_model_from_hub
-        
-        # Try to load from Hugging Face
-        model_path = hf_hub_download(
-            repo_id="PeiqingYang/MatAnyone",
-            filename="pytorch_model.bin",
-            cache_dir=self.checkpoints_dir
-        )
-        
-        model = load_model_from_hub(model_path, device=self.device)
+        """MatAnyone loading strategy 4: Direct model class"""
+        from matanyone.model.matanyone import MatAnyone
         
-        return model, model  # Return same object for both
+        model = MatAnyone.from_pretrained("not-lain/matanyone")
+        return model, model
 
 # ============================================================================ #
 # MODEL MANAGEMENT AND CLEANUP