Generate: TF can now generate from embeddings in encoder-decoder models (#21475)

2023-02-07 11:18:23 +00:00
parent 12eb528b5a
commit 1e4cf8bb44
4 changed files with 184 additions and 197 deletions
--- a/tests/generation/test_tf_utils.py
+++ b/tests/generation/test_tf_utils.py
@@ -135,6 +135,7 @@ class TFGenerationIntegrationTests(unittest.TestCase, GenerationIntegrationTests
    # setting framework_dependent_parameters needs to be gated, just like its contents' imports
    if is_tf_available():
        framework_dependent_parameters = {
+            "AutoModelForCausalLM": TFAutoModelForCausalLM,
            "AutoModelForSeq2SeqLM": TFAutoModelForSeq2SeqLM,
            "LogitsProcessorList": TFLogitsProcessorList,
            "MinLengthLogitsProcessor": TFMinLengthLogitsProcessor,