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