fixes

Author: sayakpaul

src/diffusers/utils/testing_utils.py

@@ -289,9 +289,8 @@ def decorator(test_case):
         if not torch.cuda.is_available():
             return unittest.skip(test_case)
         else:
-            compute_capability = get_torch_cuda_device_capability()
-            current_compute_capability = f"{compute_capability[0]}.{compute_capability[1]}"
-            return unittest.skipUnless(float(current_compute_capability) == float(expected_compute_capability))
+            current_compute_capability = get_torch_cuda_device_capability()
+            return unittest.skipUnless(float(current_compute_capability) == float(expected_compute_capability), "Test not supported for this compute capability.")
 
     return decorator
 

tests/quantization/quanto/test_quanto.py

@@ -313,8 +313,6 @@ def get_dummy_init_kwargs(self):
 
 
 @require_torch_cuda_compatibility(8.0)
-
-
 class FluxTransformerInt4WeightsTest(FluxTransformerQuantoMixin, unittest.TestCase):
     expected_memory_reduction = 0.55
 
@@ -323,8 +321,6 @@ def get_dummy_init_kwargs(self):
 
 
 @require_torch_cuda_compatibility(8.0)
-
-
 class FluxTransformerInt2WeightsTest(FluxTransformerQuantoMixin, unittest.TestCase):
     expected_memory_reduction = 0.65