Update audio_separator/separator/separator.py

audio_separator/separator/separator.py

@@ -582,6 +582,35 @@ class Separator:
         """
         model_path = os.path.join(self.model_file_dir, f"{model_filename}")
 
+        # BYPASS: If the model file already exists locally (pre-downloaded), skip the registry check entirely.
+        # This allows custom/non-registered models to be loaded without being in UVR's supported model list.
+        if os.path.isfile(model_path):
+            self.logger.info(f"Model file already exists at {model_path}, skipping registry lookup (bypass active)")
+            yaml_config_filename = None
+            model_basename = model_filename.rsplit('.', 1)[0]
+            # Try to find a matching YAML config file in the model directory
+            try:
+                for f in os.listdir(self.model_file_dir):
+                    if f.endswith('.yaml') and not f.startswith('model_data') and f != 'download_checks.json':
+                        # Prefer YAML files that share a name prefix with the model
+                        if model_basename in f or f.replace('.yaml', '') in model_basename:
+                            yaml_config_filename = f
+                            break
+                # If no exact match, try any YAML that was recently modified (likely downloaded together)
+                if yaml_config_filename is None:
+                    yaml_files = [f for f in os.listdir(self.model_file_dir) if f.endswith('.yaml') and not f.startswith('model_data')]
+                    if yaml_files:
+                        # Sort by modification time, newest first
+                        yaml_files.sort(key=lambda x: os.path.getmtime(os.path.join(self.model_file_dir, x)), reverse=True)
+                        yaml_config_filename = yaml_files[0]
+            except Exception as e:
+                self.logger.warning(f"Error scanning for YAML config: {e}")
+            
+            self.model_friendly_name = model_filename
+            model_type = "MDXC"  # MDXC type uses YAML-based config loading, bypassing hash check
+            self.logger.info(f"Bypass: Using model_type={model_type}, yaml_config={yaml_config_filename}")
+            return model_filename, model_type, model_filename, model_path, yaml_config_filename
+
         supported_model_files_grouped = self.list_supported_model_files()
         public_model_repo_url_prefix = "https://github.com/TRvlvr/model_repo/releases/download/all_public_uvr_models"
         vip_model_repo_url_prefix = "https://github.com/Anjok0109/ai_magic/releases/download/v5"