resolution

This commit is contained in:
505030475 2023-05-13 14:00:07 +08:00
parent 08e184ea55
commit 986653b43e

View File

@ -22,7 +22,7 @@ def gen_image(llm_kwargs, prompt, resolution="256x256"):
data = { data = {
'prompt': prompt, 'prompt': prompt,
'n': 1, 'n': 1,
'size': '256x256', 'size': resolution,
'response_format': 'url' 'response_format': 'url'
} }
response = requests.post(url, headers=headers, json=data, proxies=proxies) response = requests.post(url, headers=headers, json=data, proxies=proxies)