chansung commited on
Commit
7851c38
1 Parent(s): 0dc1378

Update palmapi.py

Browse files
Files changed (1) hide show
  1. palmapi.py +2 -4
palmapi.py CHANGED
@@ -35,16 +35,14 @@ def gen_text(
35
  parameters=None
36
  ):
37
  if parameters is None:
38
- model = 'models/text-bison-001'
39
  temperature = 0.7
40
- candidate_count = 1
41
  top_k = 40
42
  top_p = 0.95
43
 
44
  parameters = {
45
- 'model': model,
 
46
  'temperature': temperature,
47
- 'candidate_count': candidate_count,
48
  'top_k': top_k,
49
  'top_p': top_p,
50
  }
 
35
  parameters=None
36
  ):
37
  if parameters is None:
 
38
  temperature = 0.7
 
39
  top_k = 40
40
  top_p = 0.95
41
 
42
  parameters = {
43
+ 'model': 'models/text-bison-001',
44
+ 'candidate_count': 1,
45
  'temperature': temperature,
 
46
  'top_k': top_k,
47
  'top_p': top_p,
48
  }