Skip to content

GitLab

Unverified Commit 3e68944c authored by Younes Belkada's avatar Younes Belkada Committed by GitHub
Browse files

[`ClipVision`] `accelerate` support for clip-vision (#27851) 

support accelerate for clip-vision
parent b7e6d120
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
src/transformers/models/clip/modeling_clip.py
View file @ 3e68944c
...@@ -870,6 +870,7 @@ class CLIPVisionTransformer(nn.Module): ...@@ -870,6 +870,7 @@ class CLIPVisionTransformer(nn.Module):
class CLIPVisionModel(CLIPPreTrainedModel): class CLIPVisionModel(CLIPPreTrainedModel):
config_class = CLIPVisionConfig config_class = CLIPVisionConfig
main_input_name = "pixel_values" main_input_name = "pixel_values"
_no_split_modules = ["CLIPEncoderLayer"]
def __init__(self, config: CLIPVisionConfig): def __init__(self, config: CLIPVisionConfig):
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