MatteoScript commited on
Commit
13e18f4
·
verified ·
1 Parent(s): a6e1b57

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +4 -1
main.py CHANGED
@@ -146,7 +146,10 @@ async def GeneraTestoAsync(url, input_data):
146
  'temperature': input_data.temperature,
147
  'max_new_tokens': input_data.max_new_tokens,
148
  'top_p': input_data.top_p,
149
- 'repetition_penalty': input_data.repetition_penalty
 
 
 
150
  }
151
  tasks.extend([make_request(session, token, data, url) for _ in range(input_data.NumeroGenerazioni)])
152
  #tasks = [make_request(session, token, data, url) for _ in range(input_data.NumeroGenerazioni)]
 
146
  'temperature': input_data.temperature,
147
  'max_new_tokens': input_data.max_new_tokens,
148
  'top_p': input_data.top_p,
149
+ 'repetition_penalty': input_data.repetition_penalty,
150
+ 'systemRole': input_data.systemRole,
151
+ 'systemStyle': input_data.systemStyle,
152
+ 'instruction': input_data.instruction
153
  }
154
  tasks.extend([make_request(session, token, data, url) for _ in range(input_data.NumeroGenerazioni)])
155
  #tasks = [make_request(session, token, data, url) for _ in range(input_data.NumeroGenerazioni)]