fix generator 2

patrickvonplaten · patrickvonplaten · commit 5da71f8fa3e1 · 2022-06-07T16:22:12.000Z
diff --git a/tests/test_modeling_utils.py b/tests/test_modeling_utils.py
@@ -22,6 +22,7 @@
 from distutils.util import strtobool
 
 import torch
+import numpy as np
 
 from diffusers import GaussianDDPMScheduler, UNetModel
 from diffusers.pipeline_utils import DiffusionPipeline
@@ -35,7 +36,7 @@
 def get_random_generator(seed):
     seed = 1234
     random.seed(seed)
-    os.environ[‘PYTHONHASHSEED’] = str(seed)
+    os.environ["PYTHONHASHSEED"] = str(seed)
     np.random.seed(seed)
     torch.manual_seed(seed)
     torch.cuda.manual_seed(seed)
@@ -176,6 +177,7 @@ def test_sample(self):
 
         assert image.shape == (1, 3, 256, 256)
         image_slice = image[0, -1, -3:, -3:].cpu()
+        import ipdb; ipdb.set_trace()
         assert (image_slice - torch.tensor([[-0.0598, -0.0611, -0.0506], [-0.0726, 0.0220, 0.0103], [-0.0723, -0.1310, -0.2458]])).abs().sum() < 1e-3
 
     def test_sample_fast(self):
@@ -216,6 +218,7 @@ def test_sample_fast(self):
 
         assert image.shape == (1, 3, 256, 256)
         image_slice = image[0, -1, -3:, -3:].cpu()
+        import ipdb; ipdb.set_trace()
         assert (image_slice - torch.tensor([[0.1746, 0.5125, -0.7920], [-0.5734, -0.2910, -0.1984], [0.4090, -0.7740, -0.3941]])).abs().sum() < 1e-3
 
 
