Update app.py

app.py

@@ -1,6 +1,46 @@
+import sys
+import warnings
+import os
+
+warnings.filterwarnings("ignore", category=UserWarning, module="pydantic")
+
+def _monkey_patch_for_mergekit():
+    
+    import pydantic
+    import pydantic_core
+    import torch
+    
+    original_validate = pydantic_core.core_schema.is_instance_schema
+    
+    def patched_is_instance_schema(cls, *args, **kwargs):
+        if cls is torch.Tensor:
+            # Return a simple any_schema for torch.Tensor
+            return pydantic_core.core_schema.any_schema()
+        return original_validate(cls, *args, **kwargs)
+    
+    pydantic_core.core_schema.is_instance_schema = patched_is_instance_schema
+    
+    original_create_model = pydantic.create_model
+    
+    def patched_create_model(__model_name, __config__=None, **kwargs):
+        if __config__ is None:
+            __config__ = pydantic.ConfigDict(arbitrary_types_allowed=True)
+        elif isinstance(__config__, dict):
+            __config__["arbitrary_types_allowed"] = True
+        elif hasattr(__config__, "arbitrary_types_allowed"):
+            __config__.arbitrary_types_allowed = True
+        
+        return original_create_model(__model_name, __config__=__config__, **kwargs)
+    
+    pydantic.create_model = patched_create_model
+    
+    return True
+
+if _monkey_patch_for_mergekit():
+    print("Applied monkey patches for MergeKit compatibility")
+
 import gradio as gr
 import torch
-import os
 import gc
 import shutil
 import requests
@@ -15,33 +55,6 @@ from huggingface_hub import HfApi, hf_hub_download, list_repo_files, login
 from safetensors.torch import load_file, save_file
 from tqdm import tqdm
 
-import pydantic
-import torch
-import pydantic_core
-
-# Critical fix for torch.Tensor Pydantic schema
-@classmethod
-def _tensor_pydantic_schema(cls, handler):
-    from pydantic_core import core_schema
-    return core_schema.any_schema()
-
-if not hasattr(torch.Tensor, '__get_pydantic_core_schema__'):
-    torch.Tensor.__get_pydantic_core_schema__ = _tensor_pydantic_schema
-
-# Patch pydantic.create_model to always allow arbitrary types
-_original_create_model = pydantic.create_model
-
-def _patched_create_model(__model_name, __config__=None, **kwargs):
-    if __config__ is None:
-        __config__ = pydantic.ConfigDict(arbitrary_types_allowed=True)
-    elif isinstance(__config__, dict):
-        __config__ = {**__config__, "arbitrary_types_allowed": True}
-    elif hasattr(__config__, "arbitrary_types_allowed"):
-        __config__.arbitrary_types_allowed = True
-    
-    return _original_create_model(__model_name, __config__=__config__, **kwargs)
-
-pydantic.create_model = _patched_create_model
 
 # --- Memory Efficient Safetensors ---
 class MemoryEfficientSafeOpen: