Enabling imageGPT auto feature extractor. (#16871)

* Enablign `imageGPT` auto feature extractor.

Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>

* Small updates.

* Update after rebase to use `input_ids` instead of `pixel_values`.

Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>
This commit is contained in:
Nicolas Patry
2022-05-24 12:30:46 +02:00
committed by GitHub
parent 31ee80d556
commit d980929803
3 changed files with 18 additions and 2 deletions

View File

@@ -171,6 +171,12 @@ class ImageGPTModelTester:
reorder_and_upcast_attn=reorder_and_upcast_attn,
)
def get_pipeline_config(self):
config = self.get_config()
config.vocab_size = 513
config.max_position_embeddings = 1024
return config
def prepare_config_and_inputs_for_decoder(self):
(
config,