Spaces:
Running
on
T4
Running
on
T4
thomas-yanxin
commited on
Commit
·
23ba95c
1
Parent(s):
7f2c172
Update app.py
Browse files
app.py
CHANGED
@@ -23,14 +23,14 @@ embedding_model_dict = {
|
|
23 |
"ernie-tiny": "nghuyong/ernie-3.0-nano-zh",
|
24 |
"ernie-base": "nghuyong/ernie-3.0-base-zh",
|
25 |
"text2vec-base": "GanymedeNil/text2vec-base-chinese",
|
26 |
-
"ViT-B-32": 'ViT-B-32::laion2b-s34b-b79k'
|
27 |
}
|
28 |
|
29 |
llm_model_dict = {
|
30 |
"ChatGLM-6B-int8": "THUDM/chatglm-6b-int8",
|
31 |
"ChatGLM-6B-int4": "THUDM/chatglm-6b-int4",
|
32 |
"ChatGLM-6b-int4-qe": "THUDM/chatglm-6b-int4-qe",
|
33 |
-
"Minimax": "Minimax"
|
34 |
}
|
35 |
|
36 |
DEVICE = "cuda" if torch.cuda.is_available(
|
|
|
23 |
"ernie-tiny": "nghuyong/ernie-3.0-nano-zh",
|
24 |
"ernie-base": "nghuyong/ernie-3.0-base-zh",
|
25 |
"text2vec-base": "GanymedeNil/text2vec-base-chinese",
|
26 |
+
#"ViT-B-32": 'ViT-B-32::laion2b-s34b-b79k'
|
27 |
}
|
28 |
|
29 |
llm_model_dict = {
|
30 |
"ChatGLM-6B-int8": "THUDM/chatglm-6b-int8",
|
31 |
"ChatGLM-6B-int4": "THUDM/chatglm-6b-int4",
|
32 |
"ChatGLM-6b-int4-qe": "THUDM/chatglm-6b-int4-qe",
|
33 |
+
#"Minimax": "Minimax"
|
34 |
}
|
35 |
|
36 |
DEVICE = "cuda" if torch.cuda.is_available(
|