samta-kamboj commited on
Commit
08e2936
1 Parent(s): 975a8ec

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-590m"
52
  _CONFIG_FOR_DOC = "JAISConfig"
53
 
54