Spaces:

fakeym
/

travel

Sleeping

fakeym commited on Aug 18

Commit

bc9a010

•

1 Parent(s): 3479040

Update travel/self_rag_tool.py

Files changed (1) hide show

travel/self_rag_tool.py CHANGED Viewed

@@ -135,7 +135,7 @@ class embedding(object):
         self.tokenizer = AutoTokenizer.from_pretrained('maidalun1020/bce-embedding-base_v1')
         self.model = AutoModel.from_pretrained('maidalun1020/bce-embedding-base_v1')
-        self.device = 'cuda'  # if no GPU, set "cpu"
         self.model.to(self.device)

         self.tokenizer = AutoTokenizer.from_pretrained('maidalun1020/bce-embedding-base_v1')
         self.model = AutoModel.from_pretrained('maidalun1020/bce-embedding-base_v1')
+        self.device = 'cpu'  # if no GPU, set "cpu"
         self.model.to(self.device)