vtiw commited on
Commit
9ca22c6
·
verified ·
1 Parent(s): df89d31

Update utils.py

Browse files
Files changed (1) hide show
  1. utils.py +1 -1
utils.py CHANGED
@@ -174,7 +174,7 @@ import torch
174
 
175
  embedding_model_name = 'l3cube-pune/punjabi-sentence-similarity-sbert'
176
 
177
- model_kwargs = {'device':'cuda' if torch.cuda.is_available() else 'cpu',"trust_remote_code": True}
178
 
179
  embeddings = HuggingFaceEmbeddings(
180
  model_name=embedding_model_name,
 
174
 
175
  embedding_model_name = 'l3cube-pune/punjabi-sentence-similarity-sbert'
176
 
177
+ model_kwargs = {'device':'cpu',"trust_remote_code": True}
178
 
179
  embeddings = HuggingFaceEmbeddings(
180
  model_name=embedding_model_name,