Skip to content

Commit

Permalink
Format
Browse files Browse the repository at this point in the history
  • Loading branch information
mgoin committed Dec 27, 2023
1 parent 7ffb2ae commit 87c8ba5
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion tests/deepsparse/pipelines/test_clip.py
Original file line number Diff line number Diff line change
Expand Up @@ -44,7 +44,7 @@ def custom_process_inputs(self, inputs):


# This overrides the process_inputs function globally for all CLIPTextPipeline classes.
# This is needed for CLIP-ViT-B-32-256x256-DataComp-s34B-b86K since it has a second input
# This is needed for CLIP-ViT-B-32-256x256-DataComp-s34B-b86K as it has a second input
# that specifies how many tokens are present.
CLIPTextPipeline.process_inputs = custom_process_inputs

Expand Down

0 comments on commit 87c8ba5

Please sign in to comment.