theluyuan/RWKV-Runner
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

small fix

Browse Source
This commit is contained in:
josc146 2024-04-30 21:52:47 +08:00
parent 2a0ad19bc5
commit 2d853f92b9
2 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
backend-python/convert_safetensors.py vendored
View File

@ -102,6 +102,7 @@ if __name__ == "__main__":
"time_mix_w2", "time_mix_w2",
"time_decay_w1", "time_decay_w1",
"time_decay_w2", "time_decay_w2",
"lora.0",
], ],
) )
print(f"Saved to {args.output}") print(f"Saved to {args.output}")

2
backend-python/routes/completion.py
View File

@ -382,7 +382,7 @@ async def chat_completions(body: ChatCompletionBody, request: Request):
body.stop.append(f"\n\n{user_code}") body.stop.append(f"\n\n{user_code}")
body.stop.append(f"\n\n{bot_code}") body.stop.append(f"\n\n{bot_code}")
elif body.stop is None: elif body.stop is None:
body.stop = default_stop body.stop = default_stop + [f"\n\n{user_code}", f"\n\n{bot_code}"]
# if not body.presystem: # if not body.presystem:
# body.stop.append("\n\n") # body.stop.append("\n\n")