fix tests

Author: patil-suraj

tests/test_modeling_utils.py

@@ -218,9 +218,9 @@ def test_from_pretrained_save_pretrained(self):
         generator = torch.Generator()
         generator = generator.manual_seed(669472945848556)
 
-        image = ddpm(generator)
+        image = ddpm(generator=generator)
         generator = generator.manual_seed(669472945848556)
-        new_image = new_ddpm(generator)
+        new_image = new_ddpm(generator=generator)
 
         assert (image - new_image).abs().sum() < 1e-5, "Models don't give the same forward pass"
 
@@ -239,8 +239,8 @@ def test_from_pretrained_hub(self):
         generator = torch.Generator(device=torch_device)
         generator = generator.manual_seed(669472945848556)
 
-        image = ddpm(generator)
+        image = ddpm(generator=generator)
         generator = generator.manual_seed(669472945848556)
-        new_image = ddpm_from_hub(generator)
+        new_image = ddpm_from_hub(generator=generator)
 
         assert (image - new_image).abs().sum() < 1e-5, "Models don't give the same forward pass"