Enable multi-device for efficientnet (#29989)

feat: enable mult-idevice for efficientnet
huggingface · May 14, 2024 · 2fcb56a · 2fcb56a
1 parent 8d4d74b
commit 2fcb56a
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/src/transformers/models/efficientnet/modeling_efficientnet.py b/src/transformers/models/efficientnet/modeling_efficientnet.py
@@ -484,6 +484,7 @@ class EfficientNetPreTrainedModel(PreTrainedModel):
     config_class = EfficientNetConfig
     base_model_prefix = "efficientnet"
     main_input_name = "pixel_values"
+    _no_split_modules = []
 
     def _init_weights(self, module):
         """Initialize the weights"""