Spaces:
Runtime error
Runtime error
Commit ·
75e738e
1
Parent(s): f205daf
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 |
|