more

Author: sayakpaul

tests/schedulers/test_scheduler_dpm_multi.py

@@ -8,6 +8,7 @@
     DPMSolverSinglestepScheduler,
     UniPCMultistepScheduler,
 )
+import unittest
 
 from .test_schedulers import SchedulerCommonTest
 
@@ -67,6 +68,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_single.py

@@ -8,7 +8,7 @@
     DPMSolverSinglestepScheduler,
     UniPCMultistepScheduler,
 )
-
+import unittest
 from .test_schedulers import SchedulerCommonTest
 
 
@@ -65,6 +65,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_edm_dpmsolver_multistep.py

@@ -4,9 +4,9 @@
 import torch
 
 from diffusers import (
-    EDMDPMSolverMultistepScheduler,
+    EDMDPMSolverMultistepScheduler
 )
-
+import unittest
 from .test_schedulers import SchedulerCommonTest
 
 
@@ -63,6 +63,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
 
@@ -258,5 +259,6 @@ def test_duplicated_timesteps(self, **config):
             scheduler.set_timesteps(scheduler.config.num_train_timesteps)
             assert len(scheduler.timesteps) == scheduler.num_inference_steps
 
+    @unittest.skip("Test not supported.")
     def test_trained_betas(self):
         pass

tests/schedulers/test_scheduler_flax.py

@@ -15,6 +15,7 @@
 import inspect
 import tempfile
 import unittest
+import unittest
 from typing import Dict, List, Tuple
 
 from diffusers import FlaxDDIMScheduler, FlaxDDPMScheduler, FlaxPNDMScheduler
@@ -675,6 +676,7 @@ def check_over_configs(self, time_step=0, **config):
 
             assert jnp.sum(jnp.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_ipndm.py

@@ -1,7 +1,7 @@
 import tempfile
 
 import torch
-
+import unittest
 from diffusers import IPNDMScheduler
 
 from .test_schedulers import SchedulerCommonTest
@@ -50,6 +50,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_pndm.py

@@ -1,6 +1,7 @@
 import tempfile
 
 import torch
+import unittest
 
 from diffusers import PNDMScheduler
 
@@ -53,6 +54,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_unclip.py

@@ -1,6 +1,7 @@
 import torch
 
 from diffusers import UnCLIPScheduler
+import unittest
 
 from .test_schedulers import SchedulerCommonTest
 
@@ -130,8 +131,10 @@ def test_full_loop_skip_timesteps(self):
         assert abs(result_sum.item() - 258.2044983) < 1e-2
         assert abs(result_mean.item() - 0.3362038) < 1e-3
 
+    @unittest.skip("Test not supported.")
     def test_trained_betas(self):
         pass
 
+    @unittest.skip("Test not supported.")
     def test_add_noise_device(self):
         pass

tests/schedulers/test_scheduler_vq_diffusion.py

@@ -2,6 +2,7 @@
 import torch.nn.functional as F
 
 from diffusers import VQDiffusionScheduler
+import unittest
 
 from .test_schedulers import SchedulerCommonTest
 
@@ -52,5 +53,6 @@ def test_time_indices(self):
         for t in [0, 50, 99]:
             self.check_over_forward(time_step=t)
 
+    @unittest.skip("Test not supported.")
     def test_add_noise_device(self):
         pass