Update modeling_jais.py
Browse files- 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 = "
|
| 52 |
_CONFIG_FOR_DOC = "JAISConfig"
|
| 53 |
|
| 54 |
|
|
|
|
| 48 |
|
| 49 |
logger = logging.get_logger(__name__)
|
| 50 |
|
| 51 |
+
_CHECKPOINT_FOR_DOC = "inceptionai/jais-family-13b-chat"
|
| 52 |
_CONFIG_FOR_DOC = "JAISConfig"
|
| 53 |
|
| 54 |
|