MatteoScript commited on
Commit
163745b
·
verified ·
1 Parent(s): 83ed783

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +1 -2
main.py CHANGED
@@ -199,10 +199,9 @@ async def make_request(session, token, data, url, index, semaphore, max_retries=
199
  data['max_new_tokens'] = max(200, data['max_new_tokens'] - 200)
200
  else:
201
  data['max_new_tokens'] = data['max_new_tokens'] + 200
202
- timeout = aiohttp.ClientTimeout(total=1000)
203
  for _ in range(max_retries):
204
  try:
205
- async with session.post(url, headers=headers, json=data, timeout=timeout) as response:
206
  response.raise_for_status()
207
  try:
208
  result_data = await response.json()
 
199
  data['max_new_tokens'] = max(200, data['max_new_tokens'] - 200)
200
  else:
201
  data['max_new_tokens'] = data['max_new_tokens'] + 200
 
202
  for _ in range(max_retries):
203
  try:
204
+ async with session.post(url, headers=headers, json=data) as response:
205
  response.raise_for_status()
206
  try:
207
  result_data = await response.json()