Spaces:
Running
on
Zero
Running
on
Zero
update
Browse files
modular_graph_and_candidates.py
CHANGED
@@ -112,9 +112,9 @@ def embedding_similarity_clusters(models_root: Path, missing: List[str], thr: fl
|
|
112 |
|
113 |
texts = {}
|
114 |
for name in tqdm(missing, desc="Reading modeling files"):
|
115 |
-
# Skip
|
116 |
-
if
|
117 |
-
print(f"Skipping {name} (
|
118 |
continue
|
119 |
|
120 |
code = ""
|
|
|
112 |
|
113 |
texts = {}
|
114 |
for name in tqdm(missing, desc="Reading modeling files"):
|
115 |
+
# Skip models that cause GPU task aborts
|
116 |
+
if any(skip in name.lower() for skip in ["mobilebert", "lxmert"]):
|
117 |
+
print(f"Skipping {name} (causes GPU abort)")
|
118 |
continue
|
119 |
|
120 |
code = ""
|