Skip to content

Commit 28073c8

Browse files
committed
minor fix
Signed-off-by: Suguna Velury <[email protected]>
1 parent 9bdee07 commit 28073c8

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

examples/llm_qat/export.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -44,7 +44,7 @@ def get_lora_model(
4444
model = AutoModelForCausalLM.from_pretrained(ckpt_path, device_map=device_map)
4545

4646
# Restore modelopt state
47-
modelopt_state = torch.load(f"{ckpt_path}/modelopt_state_calibration.pth", weights_only=False)
47+
modelopt_state = torch.load(f"{ckpt_path}/modelopt_state_calib.pth", weights_only=False)
4848
restore_from_modelopt_state(model, modelopt_state)
4949

5050
# Restore modelopt quantizer state dict

0 commit comments

Comments
 (0)