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,