Skip to content

Commit 0bbeddc

Browse files
authored
Format code and minor bug fix (#2716)
1 parent 99d19ac commit 0bbeddc

File tree

2 files changed

+2
-1
lines changed

2 files changed

+2
-1
lines changed

fastchat/conversation.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -535,7 +535,7 @@ def get_conv_template(name: str) -> Conversation:
535535
roles=("### Instruction:", "### Response:"),
536536
sep="\n",
537537
stop_str="<|EOT|>",
538-
sep_style=SeparatorStyle.ADD_NEW_LINE_SINGLE
538+
sep_style=SeparatorStyle.ADD_NEW_LINE_SINGLE,
539539
)
540540
)
541541

fastchat/serve/inference.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -118,6 +118,7 @@ def generate_stream(
118118
token_logprobs = [None] # The first token has no logprobs.
119119
sent_interrupt = False
120120
finish_reason = None
121+
stopped = False
121122
for i in range(max_new_tokens):
122123
if i == 0: # prefill
123124
if model.config.is_encoder_decoder:

0 commit comments

Comments
 (0)