exact model name

This commit is contained in:
josc146
2023-06-19 22:30:49 +08:00
parent 967be6f88f
commit d32351c130
2 changed files with 5 additions and 3 deletions

View File

@@ -117,7 +117,7 @@ async def eval_rwkv(
yield json.dumps(
{
"response": response,
"model": "rwkv",
"model": model.name,
"choices": [
{
"delta": {"content": delta},
@@ -153,7 +153,7 @@ async def eval_rwkv(
yield json.dumps(
{
"response": response,
"model": "rwkv",
"model": model.name,
"choices": [
{
"delta": {},
@@ -173,7 +173,7 @@ async def eval_rwkv(
else:
yield {
"response": response,
"model": "rwkv",
"model": model.name,
"choices": [
{
"message": {