chrisc36 commited on
Commit
127b72a
1 Parent(s): 21ac790

Update preprocessing_molmo.py

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