Fix generate_dummy_inputs for ImageGPTOnnxConfig (huggingface#20103)

Co-authored-by: ydshieh <[email protected]>
ymwangg · Nov 7, 2022 · 2bdd9fa · 2bdd9fa
1 parent cfaeb15
commit 2bdd9fa
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/transformers/models/imagegpt/configuration_imagegpt.py b/src/transformers/models/imagegpt/configuration_imagegpt.py
@@ -197,6 +197,6 @@ def generate_dummy_inputs(
         """
 
         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