phixxy/sparkytron3000
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

fixed type hinting on negative_prompt

Browse source 
added warning on non-default negative prompt
This commit is contained in:
phixxy 2024-02-20 21:25:08 -08:00
parent 13ae53cdad
commit 0bec58412d
1 changed files with 6 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

8
cogs/stable_diffusion.py
View file

@ -298,11 +298,14 @@ class StableDiffusion(commands.Cog):
async def txt2img(self, ctx: commands.Context, prompt: str) -> None:
url = f"{self.stable_diffusion_url}/sdapi/v1/txt2img"
key_value_pairs = self.get_kv_from_ctx(ctx)
negative_prompt = self.get_negative_prompt()
if negative_prompt != self.default_neg_prompt:
ctx.send(f"Using non-default negative prompt: {negative_prompt}")
headers = {'Content-Type': 'application/json'}
payload = {
"prompt": prompt,
"steps": 25,
"negative_prompt": self.get_negative_prompt()
"negative_prompt": negative_prompt
}
if key_value_pairs:
payload.update(key_value_pairs)
@ -465,6 +468,7 @@ class StableDiffusion(commands.Cog):
await ctx.send("Command is currently disabled")
return
prompt = self.get_prompt_from_ctx(ctx)
self.logger.info(f"{ctx.author.name} used imagine. Prompt: {prompt}")
if prompt == None:
prompt = await self.generate_prompt()
@ -518,7 +522,7 @@ class StableDiffusion(commands.Cog):
help="Changes the negative prompt for imagine across all channels",
brief="Change the negative prompt for imagine"
)
async def negative_prompt(self, ctx: commands.Context, *args: list) -> None:
async def negative_prompt(self, ctx: commands.Context, *args: tuple) -> None:
message = ' '.join(args)
if not message:
message = self.default_neg_prompt