Spaces:

mishig
/

embeddings-similarity

Sleeping

mishig HF Staff commited on Sep 11, 2023

Commit

f89c622

1 Parent(s): 3d7b683

Use smaller version of the model

Files changed (1) hide show

app.py CHANGED Viewed

@@ -14,7 +14,7 @@ else:
     print("CUDA is not available. Inference on CPU.")
 seperator = "-HFSEP-"
-base_name="intfloat/e5-large-v2"
 device="cuda"
 max_length=512
 max_batch_size = 500

     print("CUDA is not available. Inference on CPU.")
 seperator = "-HFSEP-"
+base_name="intfloat/e5-small-v2"
 device="cuda"
 max_length=512
 max_batch_size = 500