docker

huggingface · Feb 7, 2024 · 582c47a · 582c47a
1 parent e6eb5a6
commit 582c47a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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():