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:
@@ -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,
|
||||
|
||||
Reference in New Issue
Block a user