We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 853fb5b commit 8806e6cCopy full SHA for 8806e6c
examples/models/llama2/export_llama_lib.py
@@ -237,8 +237,10 @@ def quantize(
237
else:
238
torch_dtype = torch.float16
239
240
- if checkpoint_path is None:
241
- checkpoint_path = Path("checkpoints/meta-llama/Llama-2-7b-chat-hf/model.pth")
+ assert checkpoint_path, "Need to specify a checkpoint"
+ assert os.path.isfile(checkpoint_path), f"{checkpoint_path} does not exist"
242
+ # if checkpoint_path is None:
243
+ # checkpoint_path = Path("checkpoints/meta-llama/Llama-2-7b-chat-hf/model.pth")
244
245
if calibration_tasks is None:
246
calibration_tasks = ["wikitext"]
0 commit comments