Spaces:
Sleeping
Sleeping
Update palmapi.py
Browse files- palmapi.py +5 -1
palmapi.py
CHANGED
|
@@ -48,8 +48,12 @@ def gen_text(
|
|
| 48 |
}
|
| 49 |
|
| 50 |
if palm is None:
|
| 51 |
-
response = palm_api.chat(messages=[prompt])
|
| 52 |
else:
|
|
|
|
|
|
|
|
|
|
|
|
|
| 53 |
response = palm.reply(prompt)
|
| 54 |
|
| 55 |
if len(response.filters) > 0 and \
|
|
|
|
| 48 |
}
|
| 49 |
|
| 50 |
if palm is None:
|
| 51 |
+
response = palm_api.chat(**parameters, messages=[prompt])
|
| 52 |
else:
|
| 53 |
+
palm.temperature = parameters['temperature']
|
| 54 |
+
palm.top_k = parameters['top_k']
|
| 55 |
+
palm.top_p = parameters['top_p']
|
| 56 |
+
|
| 57 |
response = palm.reply(prompt)
|
| 58 |
|
| 59 |
if len(response.filters) > 0 and \
|