thomas-yanxin commited on
Commit
23ba95c
·
1 Parent(s): 7f2c172

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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(