Spaces:
Runtime error
Runtime error
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(
|