ChatGPT/RefreshToV1Api
1
0
Fork 0
mirror of https://github.com/Yanyutin753/RefreshToV1Api.git synced 2025-10-18 09:04:00 +00:00
Code Issues Packages Projects Releases Wiki Activity

👀 支持传入size，控制画图接口输出图片的大小

Browse Source
This commit is contained in:
Clivia
2024-08-07 11:35:11 +08:00
parent 8a6be9013c
commit 38a10e8f2d
1 changed files with 6 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
main.py
View File

@@ -1977,12 +1977,14 @@ def images_generations():
if model not in accessible_model_list and not 'gpt-4-gizmo-' in model: if model not in accessible_model_list and not 'gpt-4-gizmo-' in model:
return jsonify({"error": "model is not accessible"}), 401 return jsonify({"error": "model is not accessible"}), 401
prompt = data.get('prompt', '')
prompt = DALLE_PROMPT_PREFIX + prompt
# 获取请求中的response_format参数，默认为"url" # 获取请求中的response_format参数，默认为"url"
response_format = data.get('response_format', 'url') response_format = data.get('response_format', 'url')
# 获取请求中的size参数，默认为"1024x1024"
response_size = data.get('size', '1024x1024')
prompt = data.get('prompt', '')
prompt = DALLE_PROMPT_PREFIX + '\nprompt:' + prompt + '\nsize:' + response_size
# stream = data.get('stream', False) # stream = data.get('stream', False)