Skip to content

Commit 1a0161a

Browse files
committed
fix: Fixed qmodel_prep qconfig_save using recipe vs fname
Signed-off-by: Brandon Groth <[email protected]>
1 parent 09c7761 commit 1a0161a

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

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)