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
......@@ -925,6 +925,7 @@ class CLIPVisionModel(CLIPPreTrainedModel):
@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)
......
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