Update configuration_florence2.py (#62)

- Update configuration_florence2.py (447e712b7902f4ad5279f9906bc16d81c9a58856)

Co-authored-by: zbing <zbing@users.noreply.huggingface.co>
This commit is contained in:
ai-modelscope
2025-08-05 15:01:02 +08:00
parent b7db1d9ea3
commit 8216ef0209
4 changed files with 30 additions and 28 deletions

View File

@ -77,7 +77,7 @@ class Florence2VisionConfig(PretrainedConfig):
>>> configuration = model.config
```"""
model_type = "florence2_vision"
model_type = "davit"
keys_to_ignore_at_inference = ["past_key_values"]
def __init__(
@ -327,7 +327,7 @@ class Florence2Config(PretrainedConfig):
self.vocab_size = vocab_size
self.projection_dim = projection_dim
if vision_config is not None:
vision_config = PretrainedConfig(**vision_config)
vision_config = Florence2VisionConfig(**vision_config)
self.vision_config = vision_config
self.vocab_size = self.vocab_size