Error out when torchao-config option is not recognized (#2107)

This commit is contained in:
Jerry Zhang
2024-11-20 17:37:28 -08:00
committed by GitHub
parent 722530fa01
commit 5c6a41facf
2 changed files with 3 additions and 1 deletions
@@ -62,6 +62,8 @@ def torchao_quantize_param_data(param: torch.Tensor, torchao_config: str):
granularity=GRANULARITY_MAP[granularity]
),
)
else:
raise ValueError(f"Unexpected config: {torchao_config}")
return dummy_linear.weight