saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-07-30 20:18:08 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

presence_mask should be sampling only (#14275)

Browse source
This commit is contained in:
Yufeng Li 2023-01-12 22:09:17 -08:00 committed by GitHub
parent cc7799835e
commit 16e39807e0
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 7 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

13
onnxruntime/python/tools/transformers/convert_generation.py
View file

@ -1537,13 +1537,14 @@ def convert_generation_model(args: argparse.Namespace, generation_type: Generati
else:
inputs.append("")
if is_sampling and args.custom and args.presence_mask:
inputs.append("presence_mask")
else:
inputs.append("")
if is_sampling:
if args.custom and args.presence_mask:
inputs.append("presence_mask")
else:
inputs.append("")
if is_sampling and args.seed:
inputs.append("seed")
if args.seed:
inputs.append("seed")
outputs = ["sequences"]
if args.output_sequences_scores: