Update src/qa.py
Browse files
src/qa.py
CHANGED
|
@@ -16,7 +16,7 @@ from transformers import AutoTokenizer, AutoModelForCausalLM, pipeline
|
|
| 16 |
print("✅ qa.py (Phi-2 Hybrid) loaded from:", __file__)
|
| 17 |
|
| 18 |
# ==========================================================
|
| 19 |
-
# 1️⃣ Cache
|
| 20 |
# ==========================================================
|
| 21 |
CACHE_DIR = "/tmp/hf_cache"
|
| 22 |
os.makedirs(CACHE_DIR, exist_ok=True)
|
|
|
|
| 16 |
print("✅ qa.py (Phi-2 Hybrid) loaded from:", __file__)
|
| 17 |
|
| 18 |
# ==========================================================
|
| 19 |
+
# 1️⃣ Cache Setups
|
| 20 |
# ==========================================================
|
| 21 |
CACHE_DIR = "/tmp/hf_cache"
|
| 22 |
os.makedirs(CACHE_DIR, exist_ok=True)
|