mmarch/text-generation-webui
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Add "seed" menu in the Parameters tab

Browse source
This commit is contained in:
wywywywy 2023-03-22 18:40:20 +00:00 • committed by GitHub
parent 45b7e53565
commit 61346b88ea
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
6 changed files with 33 additions and 19 deletions
Download patch file Download diff file Expand all files Collapse all files

6
api-example-stream.py
View file

@ -34,6 +34,7 @@ async def run(context):
'penalty_alpha': 0,
'length_penalty': 1,
'early_stopping': False,
'seed': -1,
}
session = random_hash()
@ -44,14 +45,14 @@ async def run(context):
case "send_hash":
await websocket.send(json.dumps({
"session_hash": session,
"fn_index": 9
"fn_index": 12
}))
case "estimation":
pass
case "send_data":
await websocket.send(json.dumps({
"session_hash": session,
"fn_index": 9,
"fn_index": 12,
"data": [
context,
params['max_new_tokens'],
@ -68,6 +69,7 @@ async def run(context):
params['penalty_alpha'],
params['length_penalty'],
params['early_stopping'],
params['seed'],
]
}))
case "process_starts":