Spaces:
Sleeping
Sleeping
Update nomic_fewshot.py
Browse files- nomic_fewshot.py +5 -1
nomic_fewshot.py
CHANGED
|
@@ -321,4 +321,8 @@ def build_refs_nomic(
|
|
| 321 |
embeddings.append(img_avg)
|
| 322 |
print(f" {name:<14}: {len(paths)} imgs")
|
| 323 |
|
| 324 |
-
return labels, np.stack(embeddings).astype(np.float32)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 321 |
embeddings.append(img_avg)
|
| 322 |
print(f" {name:<14}: {len(paths)} imgs")
|
| 323 |
|
| 324 |
+
return labels, np.stack(embeddings).astype(np.float32)
|
| 325 |
+
|
| 326 |
+
|
| 327 |
+
NomicTextEncoder = NomicTextEncoderONNX
|
| 328 |
+
NomicVisionEncoder = NomicVisionEncoderONNX
|