Update modeling_molmoe.py

#25

Updating modelling_molmo.py MolmoE inspired from the fix in https://huggingface.co/allenai/Molmo-7B-D-0924/discussions/41

amanrangapur changed pull request status to closed

Hey @pranay-ar , I have updated the code with few more changes, thanks for the PR.

Sign up or log in to comment