reset deterministic in tearDownClass (#11785)

* reset deterministic in tearDownClass
Signed-off-by: jiqing-feng <jiqing.feng@intel.com>

* fix deterministic setting
Signed-off-by: jiqing-feng <jiqing.feng@intel.com>

---------
Signed-off-by: jiqing-feng <jiqing.feng@intel.com>
Co-authored-by: Sayak Paul <spsayakpaul@gmail.com>

tests/quantization/bnb/test_4bit.py

@@ -98,8 +98,15 @@ class Base4bitTests(unittest.TestCase):
 
     @classmethod
     def setUpClass(cls):
+        cls.is_deterministic_enabled = torch.are_deterministic_algorithms_enabled()
+        if not cls.is_deterministic_enabled:
             torch.use_deterministic_algorithms(True)
 
+    @classmethod
+    def tearDownClass(cls):
+        if not cls.is_deterministic_enabled:
+            torch.use_deterministic_algorithms(False)
+
     def get_dummy_inputs(self):
         prompt_embeds = load_pt(
             "https://huggingface.co/datasets/hf-internal-testing/bnb-diffusers-testing-artifacts/resolve/main/prompt_embeds.pt",

tests/quantization/bnb/test_mixed_int8.py

@@ -99,8 +99,15 @@ class Base8bitTests(unittest.TestCase):
 
     @classmethod
     def setUpClass(cls):
+        cls.is_deterministic_enabled = torch.are_deterministic_algorithms_enabled()
+        if not cls.is_deterministic_enabled:
             torch.use_deterministic_algorithms(True)
 
+    @classmethod
+    def tearDownClass(cls):
+        if not cls.is_deterministic_enabled:
+            torch.use_deterministic_algorithms(False)
+
     def get_dummy_inputs(self):
         prompt_embeds = load_pt(
             "https://huggingface.co/datasets/hf-internal-testing/bnb-diffusers-testing-artifacts/resolve/main/prompt_embeds.pt",