samta-kamboj
commited on
Commit
•
08e2936
1
Parent(s):
975a8ec
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-590m"
|
52 |
_CONFIG_FOR_DOC = "JAISConfig"
|
53 |
|
54 |
|