Configurable vit parameters
#68
by
gmastrapas
- opened
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 =
|
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 |
|