Skip to content

GitLab

Unverified Commit 2bdd9fa2 authored by Yih-Dar's avatar Yih-Dar Committed by GitHub
Browse files

Fix `generate_dummy_inputs` for `ImageGPTOnnxConfig` (#20103) 


Co-authored-by: default avatarydshieh <ydshieh@users.noreply.github.com>
parent cfaeb153
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
src/transformers/models/imagegpt/configuration_imagegpt.py
View file @ 2bdd9fa2
...@@ -197,6 +197,6 @@ class ImageGPTOnnxConfig(OnnxConfig): ...@@ -197,6 +197,6 @@ class ImageGPTOnnxConfig(OnnxConfig):
""" """
input_image = self._generate_dummy_images(batch_size, num_channels, image_height, image_width) input_image = self._generate_dummy_images(batch_size, num_channels, image_height, image_width)
inputs = dict(preprocessor(input_image, framework)) inputs = dict(preprocessor(images=input_image, return_tensors=framework))
return inputs return inputs
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