aidlab/AiDA_Python
6
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat（新功能）: batch generate 增加接口说明，design batch队列修改

Browse Source 
fix（修复bug）:
docs（文档变更）:
refactor（重构）:
test(增加测试):
This commit is contained in:
zchengrong
2025-04-22 14:09:28 +08:00
parent 96002eb7f2
commit bb3ef39e10
3 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
app/service/generate_batch_image/service_batch_generate_product_image.py
View File

@@ -35,7 +35,7 @@ category = "product_image"
@celery_app.task @celery_app.task
def batch_generate_product(batch_request_data): def batch_generate_product(batch_request_data):
logger.info(f"batch_generate_product batch_request_data:{batch_request_data}") logger.info(f"batch_generate_product batch_request_data:{json.dumps(batch_request_data, indent=4)}")
tasks_id = batch_request_data['tasks_id'] tasks_id = batch_request_data['tasks_id']
user_id = tasks_id.rsplit('-', 1)[1] user_id = tasks_id.rsplit('-', 1)[1]
batch_size = batch_request_data['batch_size'] batch_size = batch_request_data['batch_size']

2
app/service/generate_batch_image/service_batch_generate_relight_image.py
View File

@@ -34,7 +34,7 @@ category = "relight_image"
@celery_app.task @celery_app.task
def batch_generate_relight(batch_request_data): def batch_generate_relight(batch_request_data):
logger.info(f"batch_generate_relight batch_request_data: {batch_request_data}") logger.info(f"batch_generate_relight batch_request_data: {json.dumps(batch_request_data, indent=4)}")
negative_prompt = 'lowres, bad anatomy, bad hands, cropped, worst quality' negative_prompt = 'lowres, bad anatomy, bad hands, cropped, worst quality'
direction = batch_request_data['direction'] direction = batch_request_data['direction']
seed = "1" seed = "1"

2
app/service/generate_batch_image/service_batch_pose_transform.py
View File

@@ -97,7 +97,7 @@ def pre_processing_image(image_url):
@celery_app.task @celery_app.task
def batch_generate_pose_transform(batch_request_data): def batch_generate_pose_transform(batch_request_data):
logger.info(f"batch_generate_pose_transform batch_request_data: {batch_request_data}") logger.info(f"batch_generate_pose_transform batch_request_data: {json.dumps(batch_request_data, indent=4)}")
batch_size = batch_request_data['batch_size'] batch_size = batch_request_data['batch_size']
image_url = batch_request_data['image_url'] image_url = batch_request_data['image_url']
image = pre_processing_image(image_url) image = pre_processing_image(image_url)