Skip to content

Commit 5371477

Browse files
committed
precommit
Signed-off-by: h-guo18 <67671475+h-guo18@users.noreply.github.com>
1 parent 57b2090 commit 5371477

1 file changed

Lines changed: 2 additions & 0 deletions

File tree

modelopt/torch/export/unified_export_hf.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1084,6 +1084,8 @@ def _unpatch_revert_weight_conversion(patches: list[tuple[Any, Any]]) -> None:
10841084
"""Restore the original revert_weight_conversion functions."""
10851085
for mod, original in patches:
10861086
mod.revert_weight_conversion = original
1087+
1088+
10871089
def export_speculative_decoding(
10881090
model: torch.nn.Module,
10891091
dtype: torch.dtype | None = None,

0 commit comments

Comments
 (0)