Spaces:
Running
on
Zero
Running
on
Zero
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:
|