duzx16 commited on
Commit
ee22604
1 Parent(s): ff8a00e

Update model variable

Browse files
Files changed (1) hide show
  1. modeling_chatglm.py +2 -2
modeling_chatglm.py CHANGED
@@ -38,11 +38,11 @@ if sys.platform != 'darwin':
38
 
39
  logger = logging.get_logger(__name__)
40
 
41
- _CHECKPOINT_FOR_DOC = "THUDM/ChatGLM2-6B"
42
  _CONFIG_FOR_DOC = "ChatGLM6BConfig"
43
 
44
  CHATGLM_6B_PRETRAINED_MODEL_ARCHIVE_LIST = [
45
- "THUDM/chatglm2-6b",
46
  # See all ChatGLM models at https://huggingface.co/models?filter=chatglm
47
  ]
48
 
 
38
 
39
  logger = logging.get_logger(__name__)
40
 
41
+ _CHECKPOINT_FOR_DOC = "THUDM/ChatGLM"
42
  _CONFIG_FOR_DOC = "ChatGLM6BConfig"
43
 
44
  CHATGLM_6B_PRETRAINED_MODEL_ARCHIVE_LIST = [
45
+ "THUDM/chatglm3-6b-base",
46
  # See all ChatGLM models at https://huggingface.co/models?filter=chatglm
47
  ]
48