Skip to content

Commit 8743c1a

Browse files
committed
fix: Fixed qmodel_prep qconfig_save using recipe vs fname
Signed-off-by: Brandon Groth <brandon.m.groth@gmail.com>
1 parent 5a68c9e commit 8743c1a

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

fms_mo/prep.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -869,7 +869,7 @@ def qmodel_prep(
869869
model, device_ids=DPorDDPdevices
870870
)
871871

872-
qconfig_save(qcfg, "qcfg.json")
872+
qconfig_save(qcfg, fname="qcfg.json")
873873
qcfg["tb_writer"] = tb_writer
874874

875875
logger.info(f"--- Quantized model --- \n{model}\n")

0 commit comments

Comments
 (0)