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 = ""
|