Molbap HF Staff commited on
Commit
ee16843
·
1 Parent(s): a093b48
Files changed (1) hide show
  1. modular_graph_and_candidates.py +1 -2
modular_graph_and_candidates.py CHANGED
@@ -126,12 +126,11 @@ def embedding_similarity_clusters(models_root: Path, missing: List[str], thr: fl
126
  texts[name] = code.strip() or " "
127
 
128
  names = list(texts)
 
129
  all_embeddings = []
130
 
131
  print(f"Encoding embeddings for {len(names)} models...")
132
  batch_size = 8
133
-
134
- names = names[:90]
135
 
136
  for i in tqdm(range(0, len(names), batch_size), desc="Models", leave=False):
137
  model_name = names[i]
 
126
  texts[name] = code.strip() or " "
127
 
128
  names = list(texts)
129
+ names = names[:90] # Limit before processing
130
  all_embeddings = []
131
 
132
  print(f"Encoding embeddings for {len(names)} models...")
133
  batch_size = 8
 
 
134
 
135
  for i in tqdm(range(0, len(names), batch_size), desc="Models", leave=False):
136
  model_name = names[i]