Skip to content

Commit 826c0c6

Browse files
authored
Update builder.py (#1387)
Wording of error message to include AOTI package
1 parent 57dee04 commit 826c0c6

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

torchchat/cli/builder.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -74,7 +74,7 @@ def __post_init__(self):
7474
or (self.pte_path and Path(self.pte_path).is_file())
7575
):
7676
raise RuntimeError(
77-
"need to specified a valid checkpoint path, checkpoint dir, gguf path, DSO path, or PTE path"
77+
"need to specified a valid checkpoint path, checkpoint dir, gguf path, DSO path, AOTI PACKAGE or PTE path"
7878
)
7979

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

0 commit comments

Comments
 (0)