Skip to content

Commit aca818a

Browse files
committed
fix:minor edit on qmodel_prep
Signed-off-by: Omobayode Fagbohungbe <[email protected]>
1 parent a2ae168 commit aca818a

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

fms_mo/prep.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -404,13 +404,13 @@ def make_quant_module(module, curr_full_name, qcfg, verbose=False):
404404
f"{curr_full_name} {type(module)} seems to be a wrapper of Linear."
405405
"Please make sure it doesn't wrap BN and activ func. Otherwise"
406406
"please create an equivalent Linear wrapper and change qcfg['mapping']."
407-
)
407+
)
408408
QLin = mapping.get(nn.Linear, None)
409409
if QLin is None:
410410
if verbose:
411411
logger.info(
412412
f"Skip quantization of {curr_full_name} - mapping of Linear is None"
413-
)
413+
)
414414
return module_output # None means no swap for this type
415415

416416
module_output = QLin(

0 commit comments

Comments
 (0)