Add _no_split_modules to CLIPModel

huggingface · Dec 29, 2023 · 8cbc978 · 8cbc978
1 parent 3cefac1
commit 8cbc978
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/src/transformers/models/clip/modeling_clip.py b/src/transformers/models/clip/modeling_clip.py
@@ -925,6 +925,7 @@ def forward(
 @add_start_docstrings(CLIP_START_DOCSTRING)
 class CLIPModel(CLIPPreTrainedModel):
     config_class = CLIPConfig
+    _no_split_modules = ["CLIPTextEmbeddings", "CLIPEncoderLayer"]
 
     def __init__(self, config: CLIPConfig):
         super().__init__(config)