Skip to content

Commit 0fa1570

Browse files
authored
Merge branch 'main' into angelayi/aoti_api_update
2 parents 06aee6c + f4ae60f commit 0fa1570

File tree

2 files changed

+3
-1
lines changed

2 files changed

+3
-1
lines changed

torchchat/generate.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1192,6 +1192,8 @@ def callback(x, *, done_generating=False):
11921192
max_seq_length=max_seq_length,
11931193
attention_backend=self.builder_args.attention_backend,
11941194
)
1195+
if generator_args.chat_mode:
1196+
start_pos += encoded.size(0)
11951197
for token_tensor, metrics in generator_func:
11961198
if token_tensor is not None:
11971199
start_pos += token_tensor.size(0)

torchchat/utils/scripts/install_utils.sh

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -88,7 +88,7 @@ install_executorch_python_libs() {
8888
echo "Building and installing python libraries"
8989
if [ "${ENABLE_ET_PYBIND}" = false ]; then
9090
echo "Not installing pybind"
91-
bash ./install_requirements.sh
91+
bash ./install_requirements.sh --pybind off
9292
else
9393
echo "Installing pybind"
9494
bash ./install_requirements.sh --pybind xnnpack

0 commit comments

Comments
 (0)