feat 新增 生成sketch时对图片清理背景,剔除带有污点的结果图

This commit is contained in:
zchen
2024-04-23 15:30:23 +08:00
parent 8f32d96a6e
commit 582e8a3dcc

View File

@@ -125,7 +125,7 @@ class GenerateImage:
inputs = [input_text, input_image, input_mode] inputs = [input_text, input_image, input_mode]
ctx = self.infer(inputs) ctx = self.infer(inputs)
time_out = 60 time_out = 600
generate_data = None generate_data = None
while time_out > 0: while time_out > 0:
generate_data, _ = self.read_tasks_status() generate_data, _ = self.read_tasks_status()