riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-07-31 02:02:21 +06:00
Code Issues Actions 33 Packages Projects Releases Wiki Activity

Fix generate_dummy_inputs for ImageGPTOnnxConfig (#20103)

Browse Source 
Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>
This commit is contained in:
Yih-Dar 2022-11-07 16:31:26 +01:00 committed by GitHub
parent cfaeb1539e
commit 2bdd9fa284
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/transformers/models/imagegpt/configuration_imagegpt.py
View File

@ -197,6 +197,6 @@ class ImageGPTOnnxConfig(OnnxConfig):
"""
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