Napron commited on
Commit
ae6bf0f
·
verified ·
1 Parent(s): 2e3c33d

Update nomic_fewshot.py

Browse files
Files changed (1) hide show
  1. 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