Update preprocessing_molmo.py
Browse files- preprocessing_molmo.py +3 -1
preprocessing_molmo.py
CHANGED
@@ -167,4 +167,6 @@ class MolmoProcessor(ProcessorMixin):
|
|
167 |
for k, v in out.items():
|
168 |
out[k] = torch.from_numpy(v)
|
169 |
|
170 |
-
return out
|
|
|
|
|
|
167 |
for k, v in out.items():
|
168 |
out[k] = torch.from_numpy(v)
|
169 |
|
170 |
+
return out
|
171 |
+
|
172 |
+
MolmoProcessor.register_for_auto_class()
|