Update main.py

main.py

@@ -10,9 +10,13 @@ import torch
 # -----------------------
 # Set cache dirs (avoid Docker errors)
 # -----------------------
-os.environ["HF_HOME"] = "/tmp"
-os.environ["TRANSFORMERS_CACHE"] = "/tmp"
+# os.environ["HF_HOME"] = "/tmp"
+# os.environ["TRANSFORMERS_CACHE"] = "/tmp"
+# os.environ["TORCHINDUCTOR_CACHE_DIR"] = "/tmp/torch_inductor_cache"
+# os.makedirs("/tmp/torch_inductor_cache", exist_ok=True)
+os.environ["TORCH_HOME"] = "/tmp/torch_home"
 os.environ["TORCHINDUCTOR_CACHE_DIR"] = "/tmp/torch_inductor_cache"
+os.makedirs("/tmp/torch_home", exist_ok=True)
 os.makedirs("/tmp/torch_inductor_cache", exist_ok=True)
 
 # -----------------------