Spaces:
Running
Running
kovacsvi
commited on
Commit
·
5ed0b8d
1
Parent(s):
654bf8b
lazy jit for dev
Browse files
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}")
|