woops
Browse files
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]
|