diff --git a/tests/models/llama/test_inference_llama.py b/tests/models/llama/test_inference_llama.py index a6fd3518..3b267470 100644 --- a/tests/models/llama/test_inference_llama.py +++ b/tests/models/llama/test_inference_llama.py @@ -26,7 +26,7 @@ def test_foo(self): model.to(device) # To make generation's sample deterministic - torch.manual_seed(0) + torch.manual_seed(1) # Generate with torch.no_grad():