lfs change to acge_text_embedding

This commit is contained in:
2024-06-04 12:29:59 +08:00
parent 4201ad34ff
commit 2bf403dca7
7 changed files with 19 additions and 1 deletions

View File

@@ -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)

BIN
models/aspire--acge_text_embedding/config.json (Stored with Git LFS) Normal file

Binary file not shown.

BIN
models/aspire--acge_text_embedding/model.safetensors (Stored with Git LFS) Normal file

Binary file not shown.

BIN
models/aspire--acge_text_embedding/special_tokens_map.json (Stored with Git LFS) Normal file

Binary file not shown.

BIN
models/aspire--acge_text_embedding/tokenizer.json (Stored with Git LFS) Normal file

Binary file not shown.

BIN
models/aspire--acge_text_embedding/tokenizer_config.json (Stored with Git LFS) Normal file

Binary file not shown.

BIN
models/aspire--acge_text_embedding/vocab.txt (Stored with Git LFS) Normal file

Binary file not shown.