From ff6b06aace5169d818acf010aa35890a9ab81804 Mon Sep 17 00:00:00 2001
From: phixxy <bottlecap@protonmail.com>
Date: Thu, 5 Oct 2023 00:57:26 -0700
Subject: [PATCH] Added lora command

---
 sparkytron3000.py | 18 ++++++++++++++++++
 1 file changed, 18 insertions(+)

diff --git a/sparkytron3000.py b/sparkytron3000.py
index cd0d65e..13ede62 100644
--- a/sparkytron3000.py
+++ b/sparkytron3000.py
@@ -1364,6 +1364,24 @@ async def change_model(ctx, model_choice='0'):
         model_options = '\n'.join([f"{choice}: {name}" for choice, name in model_choices.items()])
         output += model_options
         await ctx.send(output)
+        
+@bot.command(
+    description="Lora", 
+    help="List the stable diffusion loras.", 
+    brief="List the stable diffusion loras"
+    ) 
+async def lora(ctx):
+    lora_choices = {
+        '0': ("Lora Name", "Trigger Words"),
+        '1': ("<lora:rebecca:1>", "rebecca (cyberpunk)"),
+        '2': ("<lora:lucy:1>", "lucy (cyberpunk)"),
+        '3': ("<lora:dirty:1>", "dirty"),
+        '4': ("<lora:starcraft:1>", "c0nst3llation")
+    }
+    output = ""
+    lora_options = '\n'.join([f"{choice}: {name}" for choice, name in lora_choices.items()])
+    output += lora_options
+    await ctx.send(output)
 
 @bot.command(
     description="Imagine",