WildOjisan commited on
Commit
75e738e
·
1 Parent(s): f205daf
Files changed (1) hide show
  1. main.py +1 -0
main.py CHANGED
@@ -15,6 +15,7 @@ from transformers import (
15
  # ---- CPU 안전 설정 (run_qwen25_cpu.py에서 계승) ----
16
  os.environ.setdefault("PYTORCH_FORCE_MPS_FALLBACK", "1")
17
  os.environ.setdefault("TOKENIZERS_PARALLELISM", "false")
 
18
 
19
  MODEL_ID = "unsloth/Qwen2.5-1.5B-Instruct" # :contentReference[oaicite:3]{index=3}
20
 
 
15
  # ---- CPU 안전 설정 (run_qwen25_cpu.py에서 계승) ----
16
  os.environ.setdefault("PYTORCH_FORCE_MPS_FALLBACK", "1")
17
  os.environ.setdefault("TOKENIZERS_PARALLELISM", "false")
18
+ os.environ["HF_HOME"] = "/data"
19
 
20
  MODEL_ID = "unsloth/Qwen2.5-1.5B-Instruct" # :contentReference[oaicite:3]{index=3}
21