Molbap HF Staff commited on
Commit
661a9c6
·
1 Parent(s): de07ff6
Files changed (1) hide show
  1. modular_graph_and_candidates.py +5 -0
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: