ExternalVendorCode/lollms
1
0
Fork 0
mirror of https://github.com/ParisNeo/lollms.git synced 2024-12-20 05:08:00 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

upgraded midjourney

Browse Source
This commit is contained in:
Saifeddine ALOUI 2024-07-03 22:17:12 +02:00
parent 984c94aed6
commit d40cce3cbe
2 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
lollms/personality.py
View File

@ -618,7 +618,7 @@ class AIPersonality:
system_message_template = self.config.system_message_template system_message_template = self.config.system_message_template
prompt = "\n".join([ prompt = "\n".join([
f"{start_header_id_template}{system_message_template}{end_header_id_template}I am an AI assistant that can converse and analyze images. When asked to locate something in an image you send, I will reply with:", self.system_full_header+f"I am an AI assistant that can converse and analyze images. When asked to locate something in an image you send, I will reply with:",
"boundingbox(image_index, label, left, top, width, height)", "boundingbox(image_index, label, left, top, width, height)",
"Where:", "Where:",
"image_index: 0-based index of the image", "image_index: 0-based index of the image",

3
lollms/services/midjourney/lollms_midjourney.py
View File

@ -213,7 +213,7 @@ class LollmsMidjourney(LollmsTTI):
if width<1024: if width<1024:
file_name = self.download_image(progress_response["uri"], output_path) file_name = self.download_image(progress_response["uri"], output_path)
return file_name return file_name, {"prompt":positive_prompt, "negative_prompt":negative_prompt}
# Upscale the generated image # Upscale the generated image
upscale_response = self.upscale_image(message_id, "U1") upscale_response = self.upscale_image(message_id, "U1")
@ -230,6 +230,7 @@ class LollmsMidjourney(LollmsTTI):
return file_name, {"prompt":positive_prompt, "negative_prompt":negative_prompt} return file_name, {"prompt":positive_prompt, "negative_prompt":negative_prompt}
except Exception as e: except Exception as e:
trace_exception(e)
ASCIIColors.error(f"An error occurred: {e}") ASCIIColors.error(f"An error occurred: {e}")
@staticmethod @staticmethod