Skip to content

Commit a80e1e9

Browse files
authored
Update model.py
1 parent b218d48 commit a80e1e9

1 file changed

Lines changed: 4 additions & 4 deletions

File tree

contextplus/model.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -3,10 +3,10 @@
33
from 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 ----------------------------------------------------

0 commit comments

Comments
 (0)