ankanghosh commited on
Commit
fbf9aa9
·
verified ·
1 Parent(s): 176c1e8

Update rag_engine.py

Browse files
Files changed (1) hide show
  1. rag_engine.py +2 -2
rag_engine.py CHANGED
@@ -65,9 +65,9 @@ def load_model():
65
  device = torch.device("cpu")
66
 
67
  # Load tokenizer and model
68
- tokenizer = AutoTokenizer.from_pretrained("intfloat/e5-small-v2")
69
  model = AutoModel.from_pretrained(
70
- "intfloat/e5-small-v2",
71
  torch_dtype=torch.float16
72
  )
73
 
 
65
  device = torch.device("cpu")
66
 
67
  # Load tokenizer and model
68
+ tokenizer = AutoTokenizer.from_pretrained("intfloat/e5-large-v2")
69
  model = AutoModel.from_pretrained(
70
+ "intfloat/e5-large-v2",
71
  torch_dtype=torch.float16
72
  )
73