mirror of
https://github.com/modelscope/modelscope.git
synced 2026-09-01 19:49:03 +02:00
[to #42322933] fix generate Merge remote-tracking branch 'origin/fix_generate'
This commit is contained in:
@@ -346,3 +346,6 @@ class GPT3Model(PreTrainedModel):
|
||||
}
|
||||
model.load_state_dict(state_dict)
|
||||
return model
|
||||
|
||||
def prepare_inputs_for_generation(self, input_ids, *args, **kwargs):
|
||||
return {'input_ids': input_ids}
|
||||
|
||||
@@ -14,5 +14,4 @@ spacy>=2.3.5
|
||||
subword_nmt>=0.3.8
|
||||
text2sql_lgesql
|
||||
tokenizers
|
||||
# recent 4.23.1 update introduce breaking api change, limit upper version temporarily.
|
||||
transformers>=4.12.0,<=4.22.0
|
||||
transformers
|
||||
|
||||
Reference in New Issue
Block a user