Skip to content

Commit ed0147b

Browse files
sayakpaulDN6
andauthored
Apply suggestions from code review
Co-authored-by: Dhruv Nair <[email protected]>
1 parent cd9c0d6 commit ed0147b

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tests/quantization/quanto/test_quanto.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -312,7 +312,7 @@ def get_dummy_init_kwargs(self):
312312
return {"weights_dtype": "int8"}
313313

314314

315-
require_torch_cuda_compatibility(8.0)
315+
@require_torch_cuda_compatibility(8.0)
316316

317317

318318
class FluxTransformerInt4WeightsTest(FluxTransformerQuantoMixin, unittest.TestCase):
@@ -322,7 +322,7 @@ def get_dummy_init_kwargs(self):
322322
return {"weights_dtype": "int4"}
323323

324324

325-
require_torch_cuda_compatibility(8.0)
325+
@require_torch_cuda_compatibility(8.0)
326326

327327

328328
class FluxTransformerInt2WeightsTest(FluxTransformerQuantoMixin, unittest.TestCase):

0 commit comments

Comments
 (0)