johann22 commited on
Commit
7daae52
1 Parent(s): 0edd41f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -54,10 +54,10 @@ def run_gpt(
54
  **prompt_kwargs,
55
  ):
56
  seed = random.randint(1,1111111111111111)
57
-
58
  generate_kwargs = dict(
59
  temperature=1.0,
60
- max_new_tokens=2096,
61
  top_p=0.99,
62
  repetition_penalty=1.0,
63
  do_sample=True,
@@ -123,7 +123,7 @@ def call_main(purpose, task, history, directory, action_input):
123
  resp = run_gpt(
124
  ACTION_PROMPT,
125
  stop_tokens=["observation:", "task:"],
126
- max_tokens=2096,
127
  purpose=purpose,
128
  task=task,
129
  history=history,
 
54
  **prompt_kwargs,
55
  ):
56
  seed = random.randint(1,1111111111111111)
57
+ print (seed)
58
  generate_kwargs = dict(
59
  temperature=1.0,
60
+ max_new_tokens=1048,
61
  top_p=0.99,
62
  repetition_penalty=1.0,
63
  do_sample=True,
 
123
  resp = run_gpt(
124
  ACTION_PROMPT,
125
  stop_tokens=["observation:", "task:"],
126
+ max_tokens=1048,
127
  purpose=purpose,
128
  task=task,
129
  history=history,