Cardiosense-AG commited on
Commit
bc9a09b
·
verified ·
1 Parent(s): 8b18b46

Update src/model_loader.py

Browse files
Files changed (1) hide show
  1. 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.