lfs change to acge_text_embedding

2024-06-04 12:29:59 +08:00
parent 4201ad34ff
commit 2bf403dca7
7 changed files with 19 additions and 1 deletions
--- a/acge_embedding.py
+++ b/acge_embedding.py
@@ -6,7 +6,7 @@ import torch
 device = "cuda" if torch.cuda.is_available() else "cpu"
 print("Using device:", device)

-model_name = "models/aspire--acge-large-text"
+model_name = "models/aspire--acge_text_embedding"
 print("Loading model", model_name)
 model = (
    AutoModel.from_pretrained(model_name, local_files_only=True, torch_dtype=torch.float16).eval().to(device)