chrisc36 commited on
Commit
0e82682
1 Parent(s): d56a0b1

Update config_molmo.py

Browse files
Files changed (1) hide show
  1. 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()