From 834fcbcc30eef30a9bce0f3c4760aa2f97eed49e Mon Sep 17 00:00:00 2001 From: Jack Zhang <32371937+jackzhxng@users.noreply.github.com> Date: Fri, 16 May 2025 12:02:27 -0700 Subject: [PATCH] Fix export llava --- examples/models/llava/export_llava.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/examples/models/llava/export_llava.py b/examples/models/llava/export_llava.py index 6e0f2413786..18ef83ee1e4 100644 --- a/examples/models/llava/export_llava.py +++ b/examples/models/llava/export_llava.py @@ -111,7 +111,7 @@ def forward(self, input_pos, embeddings): quantization_mode=args.quantization_mode, group_size=args.group_size, computation_dtype=dtype_override, - checkpoint_path=args.checkpoint_path, + checkpoint_path=args.checkpoint, tokenizer_path=args.tokenizer_path, calibration_tasks=args.calibration_tasks, calibration_limit=args.calibration_limit,