Skip to content

Commit

Permalink
Update builder.py (pytorch#1387)
Browse files Browse the repository at this point in the history
Wording of error message to include AOTI package
  • Loading branch information
mikekgfb authored Nov 19, 2024
1 parent 57dee04 commit 826c0c6
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions torchchat/cli/builder.py
Original file line number Diff line number Diff line change
Expand Up @@ -74,7 +74,7 @@ def __post_init__(self):
or (self.pte_path and Path(self.pte_path).is_file())
):
raise RuntimeError(
"need to specified a valid checkpoint path, checkpoint dir, gguf path, DSO path, or PTE path"
"need to specified a valid checkpoint path, checkpoint dir, gguf path, DSO path, AOTI PACKAGE or PTE path"
)

if self.aoti_package_path and self.pte_path:
Expand All @@ -91,7 +91,7 @@ def __post_init__(self):
for param, param_msg in ignored_params:
if param:
print(
f"Warning: {param_msg} ignored because an exported DSO or PTE path was specified"
f"Warning: {param_msg} ignored because an exported model was specified using a DSO, AOTI PACKAGE or PTE path argument"
)
else:
self.prefill_possible = True
Expand Down

0 comments on commit 826c0c6

Please sign in to comment.