Skip to content

Commit

Permalink
Update optimum/intel/openvino/configuration.py
Browse files Browse the repository at this point in the history
Co-authored-by: Nikita Savelyev <[email protected]>
  • Loading branch information
l-bat and nikita-savelyevv committed Nov 13, 2024
1 parent 4a19ae7 commit ce0fc56
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion optimum/intel/openvino/configuration.py
Original file line number Diff line number Diff line change
Expand Up @@ -471,7 +471,7 @@ def post_init(self):
if self.gptq:
raise ValueError("The GPTQ algorithm is not supported for 'mxfp4' weight format")

if self.weight_format in "nf4" and self.bits != 4:
if self.weight_format == "nf4" and self.bits != 4:
raise ValueError(
f"When applying weight compression with 'nf4' weight format the `bits` parameters must be set to 4, but found {self.bits}"
)
Expand Down

0 comments on commit ce0fc56

Please sign in to comment.