Output dicts support in text generation pipeline #35092
gante
merged
8 commits into
huggingface:main
from
jonasrohw:output_dicts_support_in_text_generation_pipeline
Support for generate_argument: return_dict_in_generate=True, instead …
b6a6888e
Merge branch 'huggingface:main' into output_dicts_support_in_text_gen…
5249dd3e
fix: call test with return_dict_in_generate=True
30d595c7
fix: Only import torch if it is present
d280c0b7
update: Encapsulate output_dict changes
dc263197
fix: added back original comments
7f783ae1
Rocketknight1
approved these changes
on 2024-12-16
gante
approved these changes
on 2025-01-29
Merge branch 'main' into output_dicts_support_in_text_generation_pipe…
361910d2
Merge branch 'main' into output_dicts_support_in_text_generation_pipe…
552299de
Login to write a write a comment.
Login via GitHub