From de7f03bb587480719d9ea8bd109006e4cd6e8257 Mon Sep 17 00:00:00 2001 From: phixxy Date: Fri, 19 Jan 2024 18:55:18 -0800 Subject: [PATCH] fixed bug of neg prompt file potentially not existing --- plugins/stable_diffusion.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/plugins/stable_diffusion.py b/plugins/stable_diffusion.py index 2b1a4c2..d59cfcb 100644 --- a/plugins/stable_diffusion.py +++ b/plugins/stable_diffusion.py @@ -257,11 +257,12 @@ class StableDiffusion(commands.Cog): url=f"{url}/sdapi/v1/txt2img" prompt = ctx.message.content.split(" ", maxsplit=1)[1] key_value_pairs, prompt = self.extract_key_value_pairs(prompt) - neg_prompt_file = f"{self.db_dir}negative_prompt.txt" try: + neg_prompt_file = f"{self.db_dir}negative_prompt.txt" with open(neg_prompt_file, 'r') as f: negative_prompt = f.readline() except: + neg_prompt_file = f"{self.db_dir}negative_prompt.txt" with open(neg_prompt_file, 'w') as f: f.writelines("") negative_prompt = ""