more

Author: sayakpaul

tests/pipelines/wuerstchen/test_wuerstchen_combined.py

@@ -232,8 +232,10 @@ def test_inference_batch_single_identical(self):
     def test_float16_inference(self):
         super().test_float16_inference()
 
+    @unittest.skip(reason="Test not supported.")
     def test_callback_inputs(self):
         pass
 
+    @unittest.skip(reason="Test not supported.")
     def test_callback_cfg(self):
         pass

tests/schedulers/test_scheduler_ddim_inverse.py

@@ -1,3 +1,5 @@
+import unittest
+
 import torch
 
 from diffusers import DDIMInverseScheduler
@@ -95,6 +97,7 @@ def test_inference_steps(self):
         for t, num_inference_steps in zip([1, 10, 50], [10, 50, 500]):
             self.check_over_forward(time_step=t, num_inference_steps=num_inference_steps)
 
+    @unittest.skip("Test not supported.")
     def test_add_noise_device(self):
         pass
 

tests/schedulers/test_scheduler_deis.py

@@ -1,4 +1,5 @@
 import tempfile
+import unittest
 
 import torch
 
@@ -57,6 +58,7 @@ def check_over_configs(self, time_step=0, **config):
 
                 assert torch.sum(torch.abs(output - new_output)) < 1e-5, "Scheduler outputs are not identical"
 
+    @unittest.skip("Test not supported.")
     def test_from_save_pretrained(self):
         pass
 

tests/schedulers/test_scheduler_dpm_multi.py

@@ -1,4 +1,5 @@
 import tempfile
+import unittest
 
 import torch
 
@@ -8,7 +9,6 @@
     DPMSolverSinglestepScheduler,
     UniPCMultistepScheduler,
 )
-import unittest
 
 from .test_schedulers import SchedulerCommonTest
 

tests/schedulers/test_scheduler_dpm_single.py

@@ -1,4 +1,5 @@
 import tempfile
+import unittest
 
 import torch
 
@@ -8,7 +9,7 @@
     DPMSolverSinglestepScheduler,
     UniPCMultistepScheduler,
 )
-import unittest
+
 from .test_schedulers import SchedulerCommonTest
 
 

tests/schedulers/test_scheduler_edm_dpmsolver_multistep.py

@@ -3,10 +3,8 @@
 
 import torch
 
-from diffusers import (
-    EDMDPMSolverMultistepScheduler
-)
-import unittest
+from diffusers import EDMDPMSolverMultistepScheduler
+
 from .test_schedulers import SchedulerCommonTest
 
 

tests/schedulers/test_scheduler_flax.py

@@ -15,7 +15,6 @@
 import inspect
 import tempfile
 import unittest
-import unittest
 from typing import Dict, List, Tuple
 
 from diffusers import FlaxDDIMScheduler, FlaxDDPMScheduler, FlaxPNDMScheduler

tests/schedulers/test_scheduler_ipndm.py

@@ -1,7 +1,8 @@
 import tempfile
+import unittest
 
 import torch
-import unittest
+
 from diffusers import IPNDMScheduler
 
 from .test_schedulers import SchedulerCommonTest

tests/schedulers/test_scheduler_pndm.py

@@ -1,7 +1,7 @@
 import tempfile
+import unittest
 
 import torch
-import unittest
 
 from diffusers import PNDMScheduler
 

tests/schedulers/test_scheduler_unclip.py

@@ -1,7 +1,8 @@
+import unittest
+
 import torch
 
 from diffusers import UnCLIPScheduler
-import unittest
 
 from .test_schedulers import SchedulerCommonTest