Update app.py

app.py

@@ -555,29 +555,47 @@ def task_resize(hf_token, lora_input, new_rank, dynamic_method, dynamic_param, o
 # =================================================================================
 
 def run_mergekit_yaml(config_str, output_path, hf_token):
-    """Run mergekit-yaml CLI subprocess"""
+    """Workaround for Pydantic v2.12+ issues"""
+    # Write config to file
     config_file = TempDir / "config.yaml"
     with open(config_file, "w") as f:
         f.write(config_str)
     
+    # Create a Python script that imports mergekit with modified environment
+    script = TempDir / "run_mergekit.py"
+    script_content = f'''
+import sys
+import os
+
+# Force pydantic settings before importing mergekit
+os.environ["PYDANTIC_DISABLE_WARNING"] = "1"
+
+# Monkey patch to avoid field name warnings
+import warnings
+warnings.filterwarnings("ignore", category=UserWarning, module="pydantic")
+
+import mergekit
+from mergekit.scripts.run_yaml import main
+
+if __name__ == "__main__":
+    sys.argv = ["mergekit-yaml", "{config_file}", "{output_path}", 
+                "--allow-crimes", "--lazy-unpickle", "--copy-tokenizer"]
+    main()
+    '''
+    
+    with open(script, "w") as f:
+        f.write(script_content)
+    
     env = os.environ.copy()
     if hf_token:
         env["HF_TOKEN"] = hf_token.strip()
-
-    cmd = [
-        "mergekit-yaml",
-        str(config_file),
-        str(output_path),
-        "--allow-crimes",
-        "--lazy-unpickle",
-        "--copy-tokenizer"
-    ]
     
-    print(f"Running: {' '.join(cmd)}")
+    # Run the script
+    cmd = [sys.executable, str(script)]
     result = subprocess.run(cmd, env=env, capture_output=True, text=True)
     
     if result.returncode != 0:
-        raise RuntimeError(f"MergeKit failed:\n{result.stderr}")
+        raise RuntimeError(f"MergeKit workaround failed: {result.stderr}")
     
     return str(output_path)