File tree 1 file changed +3
-2
lines changed
docs/source/exporters/onnx/usage_guides
1 file changed +3
-2
lines changed Original file line number Diff line number Diff line change @@ -396,14 +396,13 @@ class CustomMPTOnnxConfig(TextDecoderOnnxConfig):
396
396
inputs_or_outputs[f " { name} . { i} .value " ] = {0 : " batch_size" , 2 : decoder_sequence_name}
397
397
398
398
399
- model_id = " /home/ fxmarty/hf_internship/optimum /tiny-mpt-random-remote-code"
399
+ model_id = " fxmarty/tiny-mpt-random-remote-code"
400
400
config = AutoConfig.from_pretrained(model_id, trust_remote_code = True )
401
401
402
402
onnx_config = CustomMPTOnnxConfig(
403
403
config = config,
404
404
task = " text-generation" ,
405
405
use_past_in_inputs = False ,
406
- use_present_in_outputs = True ,
407
406
)
408
407
onnx_config_with_past = CustomMPTOnnxConfig(config, task = " text-generation" , use_past = True )
409
408
@@ -419,6 +418,8 @@ main_export(
419
418
trust_remote_code = True ,
420
419
custom_onnx_configs = custom_onnx_configs,
421
420
no_post_process = True ,
421
+ legacy = True ,
422
+ opset = 14
422
423
)
423
424
```
424
425
You can’t perform that action at this time.
0 commit comments