chansung commited on
Commit
310e0d7
·
1 Parent(s): 104164a

Update palmapi.py

Browse files
Files changed (1) hide show
  1. 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 \