Init _no_split_modules to empty list to allow device_map=auto
Browse files- modeling_florence2.py +2 -0
modeling_florence2.py
CHANGED
@@ -2530,6 +2530,8 @@ class Florence2VisionModelWithProjection(Florence2PreTrainedModel):
|
|
2530 |
FLORENCE2_START_DOCSTRING,
|
2531 |
)
|
2532 |
class Florence2ForConditionalGeneration(Florence2PreTrainedModel):
|
|
|
|
|
2533 |
def __init__(self, config: Florence2Config):
|
2534 |
super().__init__(config)
|
2535 |
assert config.vision_config.model_type == 'davit', 'only DaViT is supported for now'
|
|
|
2530 |
FLORENCE2_START_DOCSTRING,
|
2531 |
)
|
2532 |
class Florence2ForConditionalGeneration(Florence2PreTrainedModel):
|
2533 |
+
_no_split_modules = []
|
2534 |
+
|
2535 |
def __init__(self, config: Florence2Config):
|
2536 |
super().__init__(config)
|
2537 |
assert config.vision_config.model_type == 'davit', 'only DaViT is supported for now'
|