Skip to content

Commit f1ba52e

Browse files
Update lora.py
Signed-off-by: Michał Marcinkiewicz <43240942+mmarcinkiewicz@users.noreply.github.com>
1 parent 2182063 commit f1ba52e

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

nemo/collections/llm/peft/lora.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -180,7 +180,7 @@ def _make_main_branch(
180180
else:
181181
raise ValueError(f"Unsupported normalization ({norm_type})")
182182
main_branch.append(op)
183-
main_branch.append(te.ops.Quantize(forward=True, backward=False))
183+
# main_branch.append(te.ops.Quantize(forward=True, backward=False))
184184

185185
# Fork to LoRA branch
186186
# Note: GEMM with beta=1 in backward pass

0 commit comments

Comments
 (0)