chunzhi/runpodsdapi
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Merge pull request #12 from daoleno/main

Browse Source 
add debug info if server encounters an error
This commit is contained in:
KL 2023-04-07 09:55:30 -07:00 committed by GitHub
commit 3f189b2ef6
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 6 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
runpod_api_test.py
View File

@ -18,6 +18,9 @@ def generation():
"input": {"prompt": prompt, "steps": 28, "negative_prompt": negative_prompt, "width": 512, "height": 768, "sampler_index": "DPM++ SDE Karras", "input": {"prompt": prompt, "steps": 28, "negative_prompt": negative_prompt, "width": 512, "height": 768, "sampler_index": "DPM++ SDE Karras",
"batch_size": 1, "seed": -1}, "batch_size": 1, "seed": -1},
}) })
if res.status_code != 200:
print(f"request failed: url: {res.url}, status code: {res.status_code}, text: {res.text}")
return
task_id = res.json()['id'] task_id = res.json()['id']
@ -26,6 +29,9 @@ def generation():
'Content-Type': 'application/json', 'Content-Type': 'application/json',
"Authorization": f"Bearer {runpod_key}" "Authorization": f"Bearer {runpod_key}"
}) })
if res.status_code != 200:
print(f"request failed: url: {res.url}, status code: {res.status_code}, text: {res.text}")
break
status = res.json()['status'] status = res.json()['status']
print(status) print(status)