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()
|