e06fe8e8c6
Secret Leaks / trufflehog (push) Failing after 1s
Build documentation / build (push) Failing after 1s
Build documentation / build_other_lang (push) Failing after 0s
CodeQL Security Analysis / CodeQL Analysis (push) Failing after 0s
PR CI / pr-ci (push) Failing after 1s
Slow tests on important models (on Push - A10) / Get all modified files (push) Failing after 1s
Slow tests on important models (on Push - A10) / Model CI (push) Has been skipped
Self-hosted runner (benchmark) / Benchmark (aws-g5-4xlarge-cache) (push) Has been cancelled
New model PR merged notification / Notify new model (push) Has been cancelled
Update Transformers metadata / build_and_package (push) Has been cancelled
2.3 KiB
2.3 KiB
Generation
各フレームワークには、それぞれの GenerationMixin クラスに実装されたテキスト生成のための Generate メソッドがあります。
- PyTorch [
~generation.GenerationMixin.generate] は [~generation.GenerationMixin] に実装されています。 - TensorFlow [
~generation.TFGenerationMixin.generate] は [~generation.TFGenerationMixin] に実装されています。 - Flax/JAX [
~generation.FlaxGenerationMixin.generate] は [~generation.FlaxGenerationMixin] に実装されています。
選択したフレームワークに関係なく、[~generation.GenerationConfig] を使用して生成メソッドをパラメータ化できます。
クラスインスタンス。動作を制御する生成パラメータの完全なリストについては、このクラスを参照してください。
生成方法のこと。
モデルの生成構成を検査する方法、デフォルトとは何か、パラメーターをアドホックに変更する方法を学習するには、 カスタマイズされた生成構成を作成して保存する方法については、「 テキスト生成戦略ガイド。このガイドでは、関連機能の使用方法についても説明しています。 トークンストリーミングのような。
GenerationConfig
autodoc generation.GenerationConfig - from_pretrained - from_model_config - save_pretrained
GenerationMixin
autodoc generation.GenerationMixin - generate - compute_transition_scores