samta-kamboj commited on
Commit
458091e
1 Parent(s): 89de3a2

Update modeling_jais.py

Browse files
Files changed (1) hide show
  1. modeling_jais.py +1 -1
modeling_jais.py CHANGED
@@ -48,7 +48,7 @@ from .configuration_jais import JAISConfig
48
 
49
  logger = logging.get_logger(__name__)
50
 
51
- _CHECKPOINT_FOR_DOC = "core42/jais-13b"
52
  _CONFIG_FOR_DOC = "JAISConfig"
53
 
54
 
 
48
 
49
  logger = logging.get_logger(__name__)
50
 
51
+ _CHECKPOINT_FOR_DOC = "inceptionai/jais-family-30b-8k"
52
  _CONFIG_FOR_DOC = "JAISConfig"
53
 
54