From 2bf403dca7b51846cfa3fe445f65e0141d473554 Mon Sep 17 00:00:00 2001 From: heimoshuiyu Date: Tue, 4 Jun 2024 12:29:59 +0800 Subject: [PATCH] lfs change to acge_text_embedding --- 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, 19 insertions(+), 1 deletion(-) create mode 100644 models/aspire--acge_text_embedding/config.json create mode 100644 models/aspire--acge_text_embedding/model.safetensors create mode 100644 models/aspire--acge_text_embedding/special_tokens_map.json create mode 100644 models/aspire--acge_text_embedding/tokenizer.json create mode 100644 models/aspire--acge_text_embedding/tokenizer_config.json create mode 100644 models/aspire--acge_text_embedding/vocab.txt diff --git a/acge_embedding.py b/acge_embedding.py index 52bfaa6..929e05a 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-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) diff --git a/models/aspire--acge_text_embedding/config.json b/models/aspire--acge_text_embedding/config.json new file mode 100644 index 0000000..ec7db06 --- /dev/null +++ b/models/aspire--acge_text_embedding/config.json @@ -0,0 +1,3 @@ +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 new file mode 100644 index 0000000..423e5a8 --- /dev/null +++ b/models/aspire--acge_text_embedding/model.safetensors @@ -0,0 +1,3 @@ +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 new file mode 100644 index 0000000..0f97119 --- /dev/null +++ b/models/aspire--acge_text_embedding/special_tokens_map.json @@ -0,0 +1,3 @@ +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 new file mode 100644 index 0000000..1195a34 --- /dev/null +++ b/models/aspire--acge_text_embedding/tokenizer.json @@ -0,0 +1,3 @@ +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 new file mode 100644 index 0000000..a89103d --- /dev/null +++ b/models/aspire--acge_text_embedding/tokenizer_config.json @@ -0,0 +1,3 @@ +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 new file mode 100644 index 0000000..0291cdf --- /dev/null +++ b/models/aspire--acge_text_embedding/vocab.txt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:45bbac6b341c319adc98a532532882e91a9cefc0329aa57bac9ae761c27b291c +size 109540