allenai
/

Molmo-7B-D-0924

Image-Text-to-Text

text-generation

Model card Files Files and versions

chrisc36 commited on Sep 25, 2024

Commit

127b72a

·

verified ·

1 Parent(s): 21ac790

Update preprocessing_molmo.py

Files changed (1) hide show

preprocessing_molmo.py +3 -1

preprocessing_molmo.py CHANGED Viewed

@@ -167,4 +167,6 @@ class MolmoProcessor(ProcessorMixin):
         for k, v in out.items():
             out[k] = torch.from_numpy(v)
-        return out

         for k, v in out.items():
             out[k] = torch.from_numpy(v)
+        return out
+    MolmoProcessor.register_for_auto_class()