Update modeling_molmo.py for compatibility
#43
by
syhuggingface
- opened
A slight modification from #41
Make code compatible with transformers==4.50.3.
Update code to align with recent changes in the Transformers library (calling _extract_past_from_model_output() raises error).
syhuggingface
changed pull request title from
Update modeling_molmo.py for compatibility with transformers v4.50.3
to Update modeling_molmo.py for compatibility
amanrangapur
changed pull request status to
merged
Thanks for the contribution.
@amanrangapur can you please this apply this change to allenai/MolmoE-1B-0924 as well? It has the exact same issue in the same spot, so should be a quick patch.