update
Browse files
modular_graph_and_candidates.py
CHANGED
|
@@ -112,6 +112,11 @@ 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 |
code = ""
|
| 116 |
for py in (models_root / name).rglob("modeling_*.py"):
|
| 117 |
try:
|
|
|
|
| 112 |
|
| 113 |
texts = {}
|
| 114 |
for name in tqdm(missing, desc="Reading modeling files"):
|
| 115 |
+
# Skip mobilebert for now to test if it's causing GPU abort
|
| 116 |
+
if "mobilebert" in name.lower():
|
| 117 |
+
print(f"Skipping {name} (mobilebert)")
|
| 118 |
+
continue
|
| 119 |
+
|
| 120 |
code = ""
|
| 121 |
for py in (models_root / name).rglob("modeling_*.py"):
|
| 122 |
try:
|