We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent cd9c0d6 commit ed0147bCopy full SHA for ed0147b
tests/quantization/quanto/test_quanto.py
@@ -312,7 +312,7 @@ def get_dummy_init_kwargs(self):
312
return {"weights_dtype": "int8"}
313
314
315
-require_torch_cuda_compatibility(8.0)
+@require_torch_cuda_compatibility(8.0)
316
317
318
class FluxTransformerInt4WeightsTest(FluxTransformerQuantoMixin, unittest.TestCase):
@@ -322,7 +322,7 @@ def get_dummy_init_kwargs(self):
322
return {"weights_dtype": "int4"}
323
324
325
326
327
328
class FluxTransformerInt2WeightsTest(FluxTransformerQuantoMixin, unittest.TestCase):
0 commit comments