microsoft
/

Florence-2-base-ft

Image-Text-to-Text

text-generation

Model card Files Files and versions Community

Xenova HF staff commited on Jun 21

Commit

e0b8f37

•

1 Parent(s): 3afa95f

Remove unnecessary deletion

Files changed (1) hide show

modeling_florence2.py +0 -1

modeling_florence2.py CHANGED Viewed

@@ -2533,7 +2533,6 @@ class Florence2ForConditionalGeneration(Florence2PreTrainedModel):
     def __init__(self, config: Florence2Config):
         super().__init__(config)
         assert config.vision_config.model_type == 'davit', 'only DaViT is supported for now'
-        del config.vision_config.model_type
         self.vision_tower = DaViT.from_config(config=config.vision_config)
         # remove unused layers
         del self.vision_tower.head

     def __init__(self, config: Florence2Config):
         super().__init__(config)
         assert config.vision_config.model_type == 'davit', 'only DaViT is supported for now'
         self.vision_tower = DaViT.from_config(config=config.vision_config)
         # remove unused layers
         del self.vision_tower.head