Skip to content

GitLab

Unverified Commit 1f5590d3 authored by Zhan Ling's avatar Zhan Ling Committed by GitHub
Browse files

Move CLIP _no_split_modules to CLIPPreTrainedModel (#27841) 

Add _no_split_modules to CLIPModel
parent a989c6c6
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
src/transformers/models/clip/modeling_clip.py
View file @ 1f5590d3
...@@ -925,6 +925,7 @@ class CLIPVisionModel(CLIPPreTrainedModel): ...@@ -925,6 +925,7 @@ class CLIPVisionModel(CLIPPreTrainedModel):
@add_start_docstrings(CLIP_START_DOCSTRING) @add_start_docstrings(CLIP_START_DOCSTRING)
class CLIPModel(CLIPPreTrainedModel): class CLIPModel(CLIPPreTrainedModel):
config_class = CLIPConfig config_class = CLIPConfig
_no_split_modules = ["CLIPTextEmbeddings", "CLIPEncoderLayer"]
def __init__(self, config: CLIPConfig): def __init__(self, config: CLIPConfig):
super().__init__(config) super().__init__(config)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment