Skip to content

GitLab

Unverified Commit dcd77ce2 authored by CyberVy's avatar CyberVy Committed by GitHub
Browse files

Fix the missing parentheses when calling is_torchao_available in quantization_config.py. (#10961) 

Update quantization_config.py
parent 11d8e3ce
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
src/diffusers/quantizers/quantization_config.py
View file @ dcd77ce2
...@@ -47,7 +47,7 @@ class QuantizationMethod(str, Enum): ...@@ -47,7 +47,7 @@ class QuantizationMethod(str, Enum):
TORCHAO = "torchao" TORCHAO = "torchao"
if is_torchao_available: if is_torchao_available():
from torchao.quantization.quant_primitives import MappingType from torchao.quantization.quant_primitives import MappingType
class TorchAoJSONEncoder(json.JSONEncoder): class TorchAoJSONEncoder(json.JSONEncoder):
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment