amazingvince
/

roastme-florence-2

Text Generation

Model card Files Files and versions Community

amazingvince commited on Oct 10, 2024

Commit

dec5a46

•

1 Parent(s): 7e534bf

Update modeling_florence2.py

Files changed (1) hide show

modeling_florence2.py +1 -1

modeling_florence2.py CHANGED Viewed

@@ -2532,7 +2532,7 @@ class Florence2VisionModelWithProjection(Florence2PreTrainedModel):
 class Florence2ForConditionalGeneration(Florence2PreTrainedModel):
     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

 class Florence2ForConditionalGeneration(Florence2PreTrainedModel):
     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