Update SwitcherAI/processors/frame/modules/face_swapper.py

SwitcherAI/processors/frame/modules/face_swapper.py

@@ -20,10 +20,10 @@ EMBEDDING_CONVERTER = None
 THREAD_LOCK = threading.Lock()
 NAME = 'FACEFUSION.FRAME_PROCESSOR.FACE_SWAPPER'
 
-# Model configurations similar to the newer version
+# Model configurations with working public URLs
 MODEL_CONFIGS = {
     'inswapper_128': {
-        'url': 'https://huggingface.co/ezioruan/inswapper_128.onnx',
+        'url': 'https://github.com/facefusion/facefusion-assets/releases/download/models/inswapper_128.onnx',
         'path': '../.assets/models/inswapper_128.onnx',
         'type': 'inswapper',
         'size': (128, 128),
@@ -32,7 +32,7 @@ MODEL_CONFIGS = {
         'requires_converter': False
     },
     'inswapper_128_fp16': {
-        'url': 'https://huggingface.co/ezioruan/inswapper_128_fp16.onnx',
+        'url': 'https://github.com/facefusion/facefusion-assets/releases/download/models/inswapper_128_fp16.onnx',
         'path': '../.assets/models/inswapper_128_fp16.onnx',
         'type': 'inswapper',
         'size': (128, 128),
@@ -41,15 +41,15 @@ MODEL_CONFIGS = {
         'requires_converter': False
     },
     'simswap_256': {
-        'url': 'https://huggingface.co/ezioruan/simswap_256.onnx',
+        'url': 'https://github.com/facefusion/facefusion-assets/releases/download/models/simswap_256.onnx',
         'path': '../.assets/models/simswap_256.onnx',
         'type': 'simswap',
         'size': (256, 256),
         'mean': [0.485, 0.456, 0.406],
         'standard_deviation': [0.229, 0.224, 0.225],
         'requires_converter': True,
-        'converter_url': 'https://huggingface.co/ezioruan/arcface_converter_simswap.onnx',
-        'converter_path': '../.assets/models/arcface_converter_simswap.onnx'
+        'converter_url': 'https://github.com/facefusion/facefusion-assets/releases/download/models/arcface_w600k_r50.onnx',
+        'converter_path': '../.assets/models/arcface_w600k_r50.onnx'
     }
 }
 
@@ -136,6 +136,40 @@ def get_frame_processor() -> Any:
                     
                 except Exception as e:
                     print(f"❌ Failed to load {model_name}: {e}")
+                    
+                    # If protobuf parsing failed, delete the corrupted file and try to re-download
+                    if "INVALID_PROTOBUF" in str(e) or "Protobuf parsing failed" in str(e):
+                        print(f"🗑️ Deleting corrupted {model_name} file...")
+                        try:
+                            if model_path_obj.exists():
+                                model_path_obj.unlink()
+                                print(f"📥 Re-downloading {model_name}...")
+                                
+                                download_directory_path = resolve_relative_path('../.assets/models')
+                                if isinstance(download_directory_path, str):
+                                    download_directory_path = Path(download_directory_path)
+                                download_directory_path.mkdir(parents=True, exist_ok=True)
+                                
+                                download_urls = [temp_config['url']]
+                                if temp_config.get('requires_converter', False):
+                                    download_urls.append(temp_config['converter_url'])
+                                
+                                conditional_download(str(download_directory_path), download_urls)
+                                
+                                # Try loading again after re-download
+                                if model_path_obj.exists() and model_path_obj.stat().st_size > 1024:
+                                    print(f"🔄 Retrying {model_name} after re-download...")
+                                    FRAME_PROCESSOR = insightface.model_zoo.get_model(
+                                        str(model_path_obj), 
+                                        providers=SwitcherAI.globals.execution_providers
+                                    )
+                                    # If successful, update the global setting and break
+                                    SwitcherAI.globals.face_swapper_model = model_name
+                                    print(f"✅ Successfully loaded face swap model: {model_name}")
+                                    break
+                        except Exception as retry_error:
+                            print(f"❌ Retry failed for {model_name}: {retry_error}")
+                    
                     continue
             
             if FRAME_PROCESSOR is None: