mirror of
https://github.com/ParisNeo/lollms-webui.git
synced 2024-12-24 14:36:39 +00:00
Added the possibility to tag stuf in the image
This commit is contained in:
parent
c55daa5c8b
commit
69d8497c66
@ -1 +1 @@
|
|||||||
Subproject commit 4206618bae32416ba576e017ac14c07b1339930a
|
Subproject commit 0bbd14e6a2ac5f8b994ea41c6bf3fecf9a24283a
|
@ -19,7 +19,7 @@ from lollms.paths import LollmsPaths
|
|||||||
from lollms.helpers import ASCIIColors, trace_exception
|
from lollms.helpers import ASCIIColors, trace_exception
|
||||||
from lollms.com import NotificationType, NotificationDisplayType, LoLLMsCom
|
from lollms.com import NotificationType, NotificationDisplayType, LoLLMsCom
|
||||||
from lollms.app import LollmsApplication
|
from lollms.app import LollmsApplication
|
||||||
from lollms.utilities import File64BitsManager, PromptReshaper, PackageManager, find_first_available_file_index, run_async, is_asyncio_loop_running, yes_or_no_input
|
from lollms.utilities import File64BitsManager, PromptReshaper, PackageManager, find_first_available_file_index, run_async, is_asyncio_loop_running, yes_or_no_input, process_ai_output
|
||||||
from lollms.generation import RECEPTION_MANAGER, ROLE_CHANGE_DECISION, ROLE_CHANGE_OURTPUT
|
from lollms.generation import RECEPTION_MANAGER, ROLE_CHANGE_DECISION, ROLE_CHANGE_OURTPUT
|
||||||
|
|
||||||
import git
|
import git
|
||||||
@ -997,6 +997,7 @@ class LOLLMSWebUI(LOLLMSElfServer):
|
|||||||
ASCIIColors.success("\nFinished executing the generation")
|
ASCIIColors.success("\nFinished executing the generation")
|
||||||
|
|
||||||
def _generate(self, prompt, n_predict, client_id, callback=None):
|
def _generate(self, prompt, n_predict, client_id, callback=None):
|
||||||
|
client = self.session.get_client(client_id)
|
||||||
self.nb_received_tokens = 0
|
self.nb_received_tokens = 0
|
||||||
self.start_time = datetime.now()
|
self.start_time = datetime.now()
|
||||||
if self.model is not None:
|
if self.model is not None:
|
||||||
@ -1017,6 +1018,18 @@ class LOLLMSWebUI(LOLLMSElfServer):
|
|||||||
n_threads=self.config['n_threads']
|
n_threads=self.config['n_threads']
|
||||||
)
|
)
|
||||||
else:
|
else:
|
||||||
|
prompt = "\n".join([
|
||||||
|
"!@>system: 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)",
|
||||||
|
"Where:",
|
||||||
|
"image_index: 0-based index of the image",
|
||||||
|
"label: brief description of what is located",
|
||||||
|
"left, top: x,y coordinates of top-left box corner (0-1 scale)",
|
||||||
|
"width, height: box dimensions as fraction of image size",
|
||||||
|
"Coordinates have origin (0,0) at top-left, (1,1) at bottom-right.",
|
||||||
|
"For other queries, I will respond conversationally to the best of my abilities.",
|
||||||
|
prompt
|
||||||
|
])
|
||||||
output = self.model.generate_with_images(
|
output = self.model.generate_with_images(
|
||||||
prompt,
|
prompt,
|
||||||
self.personality.image_files,
|
self.personality.image_files,
|
||||||
@ -1030,6 +1043,12 @@ class LOLLMSWebUI(LOLLMSElfServer):
|
|||||||
seed=self.config['seed'],
|
seed=self.config['seed'],
|
||||||
n_threads=self.config['n_threads']
|
n_threads=self.config['n_threads']
|
||||||
)
|
)
|
||||||
|
try:
|
||||||
|
post_processed_output = process_ai_output(output, self.personality.image_files, client.discussion.discussion_folder)
|
||||||
|
if len(post_processed_output)!=output:
|
||||||
|
self.process_chunk(post_processed_output, MSG_TYPE.MSG_TYPE_FULL,client_id=client_id)
|
||||||
|
except Exception as ex:
|
||||||
|
ASCIIColors.error(str(ex))
|
||||||
else:
|
else:
|
||||||
ASCIIColors.info(f"warmup for generating up to {n_predict} tokens")
|
ASCIIColors.info(f"warmup for generating up to {n_predict} tokens")
|
||||||
if self.config["override_personality_model_parameters"]:
|
if self.config["override_personality_model_parameters"]:
|
||||||
|
@ -1 +1 @@
|
|||||||
Subproject commit 2ac2075cdc0685eee30983e3ba433e8e46b4386f
|
Subproject commit 69973550700ef945c1d74509fd20d4af5ff75488
|
Loading…
Reference in New Issue
Block a user