From 87ac0318594486cd9df8027298d920a498537d0a Mon Sep 17 00:00:00 2001 From: Jaret Burkett Date: Thu, 13 Feb 2025 08:42:48 -0700 Subject: [PATCH] Remove system prompt, shouldnt be necessary fo rhow it works. --- toolkit/models/llm_adapter.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/toolkit/models/llm_adapter.py b/toolkit/models/llm_adapter.py index 7efcd913..8a77eefe 100644 --- a/toolkit/models/llm_adapter.py +++ b/toolkit/models/llm_adapter.py @@ -45,7 +45,8 @@ class LLMAdapter(torch.nn.Module): self.llm_ref: weakref.ref = weakref.ref(llm) self.tokenizer_ref: weakref.ref = weakref.ref(tokenizer) - self.system_prompt = "You are an assistant designed to generate superior images with the superior degree of image-text alignment based on textual prompts or user prompts." + self.system_prompt = "" + # self.system_prompt = "You are an assistant designed to generate superior images with the superior degree of image-text alignment based on textual prompts or user prompts. " self.hidden_size = llm.config.hidden_size @@ -109,7 +110,8 @@ class LLMAdapter(torch.nn.Module): prompt = prompt if isinstance(prompt, list) else [prompt] - prompt = [self.system_prompt + " " + p for p in prompt] + # prompt = [self.system_prompt + " " + p for p in prompt] + prompt = [self.system_prompt + p for p in prompt] prompt_embeds, prompt_attention_mask = self._get_prompt_embeds( prompt=prompt,