Skip to content

Commit 7510770

Browse files
committed
fix: correcting lint error
Signed-off-by: Omobayode Fagbohungbe <[email protected]>
1 parent d3e7c61 commit 7510770

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

fms_mo/utils/dq_inf.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -157,7 +157,7 @@ def update_qcfg_from_model_config(
157157
"weights"
158158
]["num_bits"]
159159
qcfg["torch_dtype"] = "float16"
160-
if config["quantization_config"]["ignore"] is not []:
160+
if config["quantization_config"]["ignore"] != []:
161161
qcfg["qskip_layer_name"] = config["quantization_config"]["ignore"]
162162
qcfg["qskip_large_mag_layers"] = True
163163
else:
@@ -201,7 +201,7 @@ def update_config(
201201
data["quantization_config"]["config_groups"]["group_0"]["weights"] = (
202202
"{num_bits: 8, type: float, symmetric: true, strategy: tensor}"
203203
)
204-
if qcfg["qskip_large_mag_layers"] == True:
204+
if qcfg["qskip_large_mag_layers"] is True:
205205
data["quantization_config"]["ignore"] = qcfg["qskip_layer_name"]
206206
model_config_file.update(data)
207207
return model_config_file

fms_mo/utils/import_utils.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -42,7 +42,7 @@
4242
"torchvision",
4343
"huggingface_hub",
4444
"torchao",
45-
"compressed_tensors"
45+
"compressed_tensors",
4646
]
4747

4848
available_packages = {}

0 commit comments

Comments
 (0)