cmarkea
/

bloomz-3b-retriever

Feature Extraction

text-generation-inference

Inference Endpoints

Model card Files Files and versions Community

Cyrile commited on Nov 2, 2023

Commit

da851f5

·

1 Parent(s): 9f34c3a

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -13,7 +13,7 @@ import numpy as np
 from transformers import pipeline
 from scipy.spatial.distance import cdist
-retriever = pipeline('feature-extraction', 'cmarkea/bloomz-560m-retriever')
 infer = lambda x: [ii[0][-1] for ii in retriever(x)]
 list_of_contexts = [...]

 from transformers import pipeline
 from scipy.spatial.distance import cdist
+retriever = pipeline('feature-extraction', 'cmarkea/bloomz-3b-retriever')
 infer = lambda x: [ii[0][-1] for ii in retriever(x)]
 list_of_contexts = [...]