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

feat

Browse Source 
fix generate image sketch 和prompt 组合
This commit is contained in:
zhouchengrong
2024-06-28 15:47:47 +08:00
parent f44e929b33
commit 1b70786784
1 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
app/service/generate_image/service_generate_image.py
View File

@@ -40,16 +40,17 @@ class GenerateImage:
if request_data.mode == "img2img": if request_data.mode == "img2img":
# cv2 读图片是BGR PIL读图片是RGB # cv2 读图片是BGR PIL读图片是RGB
self.image = self.get_image(request_data.image_url) self.image = self.get_image(request_data.image_url)
self.prompt = request_data.prompt
else: else:
self.image = np.random.randint(0, 256, (1024, 1024, 3), dtype=np.uint8) self.image = np.random.randint(0, 256, (1024, 1024, 3), dtype=np.uint8)
self.prompt = request_data.prompt
self.prompt = request_data.prompt
self.tasks_id = request_data.tasks_id self.tasks_id = request_data.tasks_id
self.user_id = self.tasks_id[self.tasks_id.rfind('-') + 1:] self.user_id = self.tasks_id[self.tasks_id.rfind('-') + 1:]
self.mode = request_data.mode self.mode = request_data.mode
self.batch_size = 1 self.batch_size = 1
self.category = request_data.category self.category = request_data.category
if self.category == "sketch":
self.prompt = f"{self.category},{self.prompt}"
self.index = 0 self.index = 0
self.gender = request_data.gender self.gender = request_data.gender
self.generate_data = {'tasks_id': self.tasks_id, 'status': 'PENDING', 'message': "pending", 'image_url': '', 'category': ''} self.generate_data = {'tasks_id': self.tasks_id, 'status': 'PENDING', 'message': "pending", 'image_url': '', 'category': ''}