fix: patch getpwuid error for HF Spaces

train.py

@@ -1,4 +1,23 @@
-import os, sys, json, re, threading, pathlib
+import os
+import pwd
+import getpass
+
+# Fix getpwuid error in HF Spaces
+os.environ["TORCHINDUCTOR_CACHE_DIR"] = "/tmp/torch_cache"
+os.environ["TRITON_CACHE_DIR"] = "/tmp/triton_cache"
+os.makedirs("/tmp/torch_cache", exist_ok=True)
+os.makedirs("/tmp/triton_cache", exist_ok=True)
+
+# Patch getpwuid
+try:
+    pwd.getpwuid(os.getuid())
+except KeyError:
+    import ctypes
+    import ctypes.util
+    # Override getuser to return a safe default
+    getpass.getuser = lambda: "trainer"
+
+import sys, json, re, threading, pathlib
 from http.server import HTTPServer, BaseHTTPRequestHandler
 
 os.environ["HF_HOME"] = "/tmp/huggingface"