Update src/model_loader.py
Browse files- src/model_loader.py +5 -0
src/model_loader.py
CHANGED
|
@@ -14,6 +14,11 @@ from transformers import (
|
|
| 14 |
)
|
| 15 |
from accelerate import init_empty_weights, load_checkpoint_and_dispatch
|
| 16 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 17 |
HF_CACHE = os.environ.get("HF_HOME") or os.environ.get("TRANSFORMERS_CACHE") or "/data/econsult/hf_cache"
|
| 18 |
|
| 19 |
# Accept MODEL_ID (preferred) or fallback to MODEL_PRIMARY_ID to avoid env-name drift.
|
|
|
|
| 14 |
)
|
| 15 |
from accelerate import init_empty_weights, load_checkpoint_and_dispatch
|
| 16 |
|
| 17 |
+
# --- Diagnostic print to confirm runtime versions ---
|
| 18 |
+
import transformers
|
| 19 |
+
print("[init]", "torch", torch.__version__, "transformers", transformers.__version__)
|
| 20 |
+
|
| 21 |
+
|
| 22 |
HF_CACHE = os.environ.get("HF_HOME") or os.environ.get("TRANSFORMERS_CACHE") or "/data/econsult/hf_cache"
|
| 23 |
|
| 24 |
# Accept MODEL_ID (preferred) or fallback to MODEL_PRIMARY_ID to avoid env-name drift.
|