feat(新功能): batch generate 增加接口说明,design batch队列修改
fix(修复bug): docs(文档变更): refactor(重构): test(增加测试):
This commit is contained in:
36
app/service/generate_batch_image/test.py
Normal file
36
app/service/generate_batch_image/test.py
Normal file
@@ -0,0 +1,36 @@
|
||||
from app.schemas.generate_image import BatchGenerateRelightImageModel, BatchGenerateProductImageModel
|
||||
from app.service.generate_batch_image.service_batch_generate_product_image import batch_generate_product
|
||||
|
||||
from app.service.generate_batch_image.service_batch_generate_relight_image import batch_generate_relight
|
||||
|
||||
if __name__ == '__main__':
|
||||
rd = BatchGenerateProductImageModel(
|
||||
tasks_id="test1-89",
|
||||
image_strength=0.7,
|
||||
prompt=" The best quality, masterpiece, real image.Outwear,high quality clothing details,8K realistic,HDR",
|
||||
image_url="aida-results/result_40b1a2fe-e220-11ef-9bfa-0242ac150003.png",
|
||||
product_type="single",
|
||||
batch_size=2
|
||||
)
|
||||
x = batch_generate_product.delay(rd.dict())
|
||||
print(x)
|
||||
|
||||
"""relight"""
|
||||
# rd = BatchGenerateRelightImageModel(
|
||||
# tasks_id="123-89",
|
||||
# # prompt="beautiful woman, detailed face, sunshine, outdoor, warm atmosphere",
|
||||
# prompt="Colorful black",
|
||||
# image_url='aida-results/result_0000b606-1902-11ef-9424-0242ac180002.png',
|
||||
# direction="Right Light",
|
||||
# product_type="single",
|
||||
# batch_size=2
|
||||
# )
|
||||
# batch_generate_relight.delay(rd.dict())
|
||||
"""pose transform"""
|
||||
# rd = BatchPoseTransformModel(
|
||||
# tasks_id="123-89",
|
||||
# image_url='aida-results/result_0000b606-1902-11ef-9424-0242ac180002.png',
|
||||
# pose_id="1",
|
||||
# batch_size=10
|
||||
# )
|
||||
# batch_pose_transform_tasks.delay(rd.dict())
|
||||
Reference in New Issue
Block a user