Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -117,11 +117,11 @@ with gr.Blocks() as demo:
|
|
117 |
data = {
|
118 |
"model": MODEL_NAME,
|
119 |
"prompt": str(message),
|
120 |
-
"temperature": temperature,
|
121 |
-
"n": top_k,
|
122 |
-
"max_tokens": max_new_tokens,
|
123 |
"stop": "",
|
124 |
-
"top_p": top_p,
|
125 |
"logprobs": 0,
|
126 |
"echo": False,
|
127 |
"presence_penalty": PRESENCE_PENALTY,
|
|
|
117 |
data = {
|
118 |
"model": MODEL_NAME,
|
119 |
"prompt": str(message),
|
120 |
+
"temperature": float(temperature),
|
121 |
+
"n": int(top_k),
|
122 |
+
"max_tokens": int(max_new_tokens),
|
123 |
"stop": "",
|
124 |
+
"top_p": float(top_p),
|
125 |
"logprobs": 0,
|
126 |
"echo": False,
|
127 |
"presence_penalty": PRESENCE_PENALTY,
|