allenai
/

Molmo-7B-O-0924

Image-Text-to-Text

text-generation

Model card Files Files and versions

Update modeling_molmo.py

#18

by Howuhh - opened Jun 19

base: refs/heads/main

←

from: refs/pr/18

Discussion Files changed

Files changed (1) hide show

modeling_molmo.py +0 -1

modeling_molmo.py CHANGED Viewed

@@ -2272,7 +2272,6 @@ class MolmoForCausalLM(PreTrainedModel):
                 del model_kwargs["images"]
                 del model_kwargs["image_masks"]
                 del model_kwargs["image_input_idx"]
-        cache_name, cache = super()._extract_past_from_model_output(outputs)
         try:
             cache_name, cache = super()._extract_past_from_model_output(outputs)
         except AttributeError:

                 del model_kwargs["images"]
                 del model_kwargs["image_masks"]
                 del model_kwargs["image_input_idx"]
         try:
             cache_name, cache = super()._extract_past_from_model_output(outputs)
         except AttributeError: