File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff line change 33from transformers import pipeline
44
55# Loading models
6- device = "cpu" # todo only for cpu testing, can be removed to automatically choose the device
7- gist_embedding = SentenceTransformer ("avsolatorio/GIST-small-Embedding-v0" , device = device )
8- bart_summarizer = pipeline ("summarization" , model = "facebook/bart-large-cnn" , device = device )
9- flan_t5 = pipeline ("text2text-generation" , model = "google/flan-t5-base" , device = device )
6+ # device = "cpu" # todo only for cpu testing, can be removed to automatically choose the device
7+ gist_embedding = SentenceTransformer ("avsolatorio/GIST-small-Embedding-v0" )
8+ bart_summarizer = pipeline ("summarization" , model = "facebook/bart-large-cnn" )
9+ flan_t5 = pipeline ("text2text-generation" , model = "google/flan-t5-base" )
1010
1111
1212# ------------------------------------------------ Embedding Model ----------------------------------------------------
You can’t perform that action at this time.
0 commit comments