diff --git a/acge_embedding.py b/acge_embedding.py index 929e05a..52bfaa6 100644 --- 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_text_embedding" +model_name = "models/aspire--acge-large-text" print("Loading model", model_name) model = ( AutoModel.from_pretrained(model_name, local_files_only=True, torch_dtype=torch.float16).eval().to(device) diff --git a/models/aspire--acge_text_embedding/config.json b/models/aspire--acge_text_embedding/config.json deleted file mode 100644 index ec7db06..0000000 --- a/models/aspire--acge_text_embedding/config.json +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:84f6185a27968a5140eda6442e9c8046edecf7ec4a25e0304c2a1d75033eea6d -size 812 diff --git a/models/aspire--acge_text_embedding/model.safetensors b/models/aspire--acge_text_embedding/model.safetensors deleted file mode 100644 index 423e5a8..0000000 --- a/models/aspire--acge_text_embedding/model.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:c6ec3d802bff018c0ee48be067f60aae32a1d444d214663ee6888f53b6317b09 -size 652146704 diff --git a/models/aspire--acge_text_embedding/special_tokens_map.json b/models/aspire--acge_text_embedding/special_tokens_map.json deleted file mode 100644 index 0f97119..0000000 --- a/models/aspire--acge_text_embedding/special_tokens_map.json +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:b6d346be366a7d1d48332dbc9fdf3bf8960b5d879522b7799ddba59e76237ee3 -size 125 diff --git a/models/aspire--acge_text_embedding/tokenizer.json b/models/aspire--acge_text_embedding/tokenizer.json deleted file mode 100644 index 1195a34..0000000 --- a/models/aspire--acge_text_embedding/tokenizer.json +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:7dfbf1966ebf99d471c3796e9b457329d2b2182b817e144f1e904b957745c839 -size 439124 diff --git a/models/aspire--acge_text_embedding/tokenizer_config.json b/models/aspire--acge_text_embedding/tokenizer_config.json deleted file mode 100644 index a89103d..0000000 --- a/models/aspire--acge_text_embedding/tokenizer_config.json +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:644a5d53f60d2788d07caedfa54273574a84c1d30df06160a19d5a70d4073d02 -size 367 diff --git a/models/aspire--acge_text_embedding/vocab.txt b/models/aspire--acge_text_embedding/vocab.txt deleted file mode 100644 index 0291cdf..0000000 --- a/models/aspire--acge_text_embedding/vocab.txt +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:45bbac6b341c319adc98a532532882e91a9cefc0329aa57bac9ae761c27b291c -size 109540