From 88f1bb8b649708cbeb128be0a19591c8ebaa6c6f Mon Sep 17 00:00:00 2001 From: heimoshuiyu Date: Tue, 4 Jun 2024 14:06:40 +0800 Subject: [PATCH] Revert "lfs change to acge_text_embedding" This reverts commit 2bf403dca7b51846cfa3fe445f65e0141d473554. --- acge_embedding.py | 2 +- models/aspire--acge_text_embedding/config.json | 3 --- models/aspire--acge_text_embedding/model.safetensors | 3 --- models/aspire--acge_text_embedding/special_tokens_map.json | 3 --- models/aspire--acge_text_embedding/tokenizer.json | 3 --- models/aspire--acge_text_embedding/tokenizer_config.json | 3 --- models/aspire--acge_text_embedding/vocab.txt | 3 --- 7 files changed, 1 insertion(+), 19 deletions(-) delete mode 100644 models/aspire--acge_text_embedding/config.json delete mode 100644 models/aspire--acge_text_embedding/model.safetensors delete mode 100644 models/aspire--acge_text_embedding/special_tokens_map.json delete mode 100644 models/aspire--acge_text_embedding/tokenizer.json delete mode 100644 models/aspire--acge_text_embedding/tokenizer_config.json delete mode 100644 models/aspire--acge_text_embedding/vocab.txt 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