bupa1018 commited on
Commit
28bee07
·
verified ·
1 Parent(s): be33cff

Update embeddings.py

Browse files
Files changed (1) hide show
  1. embeddings.py +0 -3
embeddings.py CHANGED
@@ -15,9 +15,6 @@ def get_hf_embeddings(model_name=None):
15
  def get_SFR_Code_embedding_model(
16
  model_name="Salesforce/SFR-Embedding-Code-400M_R", device="auto"
17
  ):
18
- """Get jinaai embedding."""
19
-
20
- # device: cpu or cuda
21
  if device == "auto":
22
  device = "cuda" if torch.cuda.is_available() else "cpu"
23
 
 
15
  def get_SFR_Code_embedding_model(
16
  model_name="Salesforce/SFR-Embedding-Code-400M_R", device="auto"
17
  ):
 
 
 
18
  if device == "auto":
19
  device = "cuda" if torch.cuda.is_available() else "cpu"
20