kovacsvi commited on
Commit
5ed0b8d
·
1 Parent(s): 654bf8b

lazy jit for dev

Browse files
Files changed (1) hide show
  1. utils.py +1 -1
utils.py CHANGED
@@ -74,7 +74,7 @@ def download_hf_models():
74
  safe_model_name = model_id.replace("/", "_")
75
  traced_model_path = os.path.join(JIT_DIR, f"{safe_model_name}.pt")
76
 
77
- if os.path.exists(traced_model_path):
78
  print(f"⏩ Skipping JIT — already exists: {traced_model_path}")
79
  else:
80
  print(f"⚙️ Tracing and saving: {traced_model_path}")
 
74
  safe_model_name = model_id.replace("/", "_")
75
  traced_model_path = os.path.join(JIT_DIR, f"{safe_model_name}.pt")
76
 
77
+ if os.path.exists(traced_model_path) or "pooled-cap" not in model_id:
78
  print(f"⏩ Skipping JIT — already exists: {traced_model_path}")
79
  else:
80
  print(f"⚙️ Tracing and saving: {traced_model_path}")