Update config_molmo.py
Browse files- config_molmo.py +3 -1
config_molmo.py
CHANGED
|
@@ -54,4 +54,6 @@ class MolmoConfig(PretrainedConfig):
|
|
| 54 |
super().__init__(
|
| 55 |
tie_word_embeddings=tie_word_embeddings,
|
| 56 |
**kwargs,
|
| 57 |
-
)
|
|
|
|
|
|
|
|
|
| 54 |
super().__init__(
|
| 55 |
tie_word_embeddings=tie_word_embeddings,
|
| 56 |
**kwargs,
|
| 57 |
+
)
|
| 58 |
+
|
| 59 |
+
MolmoConfig.register_for_auto_class()
|