[CI][BugFix] Flip is_quant_method_supported condition (#5577)

This commit is contained in:
Michael Goin
2024-06-16 14:07:34 +00:00
committed by GitHub
parent f31c1f90e3
commit 4a6769053a
+1 -1
View File
@@ -10,5 +10,5 @@ def is_quant_method_supported(quant_method: str) -> bool:
capability = torch.cuda.get_device_capability()
capability = capability[0] * 10 + capability[1]
return (capability <
return (capability >=
QUANTIZATION_METHODS[quant_method].get_min_capability())