Revert "lfs change to acge_text_embedding"

This reverts commit 2bf403dca7.
This commit is contained in:
2024-06-04 14:06:40 +08:00
parent c4d65ebcfd
commit 88f1bb8b64
7 changed files with 1 additions and 19 deletions

View File

@@ -6,7 +6,7 @@ import torch
device = "cuda" if torch.cuda.is_available() else "cpu" device = "cuda" if torch.cuda.is_available() else "cpu"
print("Using device:", device) print("Using device:", device)
model_name = "models/aspire--acge_text_embedding" model_name = "models/aspire--acge-large-text"
print("Loading model", model_name) print("Loading model", model_name)
model = ( model = (
AutoModel.from_pretrained(model_name, local_files_only=True, torch_dtype=torch.float16).eval().to(device) AutoModel.from_pretrained(model_name, local_files_only=True, torch_dtype=torch.float16).eval().to(device)

Binary file not shown.

Binary file not shown.

Binary file not shown.

Binary file not shown.

Binary file not shown.

Binary file not shown.