Configurable vit parameters

#68
Files changed (1) hide show
  1. configuration_florence2.py +1 -1
configuration_florence2.py CHANGED
@@ -327,7 +327,7 @@ class Florence2Config(PretrainedConfig):
327
  self.vocab_size = vocab_size
328
  self.projection_dim = projection_dim
329
  if vision_config is not None:
330
- vision_config = PretrainedConfig(**vision_config)
331
  self.vision_config = vision_config
332
  self.vocab_size = self.vocab_size
333
 
 
327
  self.vocab_size = vocab_size
328
  self.projection_dim = projection_dim
329
  if vision_config is not None:
330
+ vision_config = Florence2VisionConfig(**vision_config)
331
  self.vision_config = vision_config
332
  self.vocab_size = self.vocab_size
333