From 2fcb56a31a52924119bd9251edd3ce645eb5948e Mon Sep 17 00:00:00 2001
From: Jacky Lee <39754370+jla524@users.noreply.github.com>
Date: Wed, 3 Apr 2024 12:54:34 -0700
Subject: [PATCH] Enable multi-device for efficientnet (#29989)

feat: enable mult-idevice for efficientnet
---
 src/transformers/models/efficientnet/modeling_efficientnet.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/src/transformers/models/efficientnet/modeling_efficientnet.py b/src/transformers/models/efficientnet/modeling_efficientnet.py
index 5b7ff534eedfe4..e415d7f1b46a1e 100644
--- 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"""