🔥py38 + torch 2 🔥🔥🔥🚀 (#22204)

* py38 + torch 2 * increment cache versions --------- Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>
2023-03-16 22:59:23 +01:00
parent fb366b9a2a
commit 5110e5748e
6 changed files with 22 additions and 14 deletions
--- a/tests/generation/test_utils.py
+++ b/tests/generation/test_utils.py
@@ -2450,7 +2450,7 @@ class GenerationIntegrationTests(unittest.TestCase, GenerationIntegrationTestsMi
            "top_k": 10,
            "temperature": 0.7,
        }
-        expectation = 15
+        expectation = 20

        tokenizer = AutoTokenizer.from_pretrained("hf-internal-testing/tiny-random-gpt2")
        text = """Hello, my dog is cute and"""