lfs add acge-large-text

acge_embedding.py

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