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