Enable multi-device for efficientnet (#29989)

feat: enable mult-idevice for efficientnet
huggingface · Apr 22, 2024 · b9a19a6 · b9a19a6
1 parent 91291b5
commit b9a19a6
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"""