Closed marcromeyn closed 1 year ago
Triton doesn't support models that output a dictionary. This PR introduces modules to convert a dict to a tuple, this can be used inside the save_for_triton function that will be added soon.
save_for_triton
https://nvidia-merlin.github.io/models/review/pr-1178
github-actions[bot]
commented
1 year ago
Goals :soccer:
Triton doesn't support models that output a dictionary. This PR introduces modules to convert a dict to a tuple, this can be used inside the
save_for_tritonfunction that will be added soon.