yayayaaa commited on
Commit
7f0a58b
1 Parent(s): e018bec

Init _no_split_modules to empty list to allow device_map=auto

Browse files
Files changed (1) hide show
  1. 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'