Skip to content

Commit 545e1d0

Browse files
committed
generate_from_hub
1 parent acde30f commit 545e1d0

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/transformers/generation/utils.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2126,7 +2126,7 @@ def generate(
21262126
- [`~generation.GenerateBeamEncoderDecoderOutput`]
21272127
"""
21282128
# 0. If requested, load an arbitrary generation recipe from the Hub and run it instead
2129-
if recipe is not None:
2129+
if recipe is not None: # TODO: should also go in the generation config, so that users can save it
21302130
# Get all `generate` arguments in a single variable. Custom functions are responsible for handling them:
21312131
# they receive the same inputs as `generate`, only with `model` instead of `self`. They can access to
21322132
# methods from `GenerationMixin` through `model`.

0 commit comments

Comments
 (0)