kojii

edenartlab · Dec 14, 2023 · d1ee3d7 · d1ee3d7
1 parent f3d654a
commit d1ee3d7
Show file tree

Hide file tree

Showing 3 changed files with 15 additions and 42 deletions.
diff --git a/src/bots/kojii/KojiiAssistantCog.py b/src/bots/kojii/KojiiAssistantCog.py
diff --git a/src/bots/kojii/KojiiCharacterCog.py b/src/bots/kojii/KojiiCharacterCog.py
@@ -0,0 +1,11 @@
+from discord.ext import commands
+from cogs.CharacterCog import CharacterCog
+
+
+class KojiiCharacterCog(CharacterCog):
+    def __init__(self, bot: commands.bot) -> None:
+        super().__init__(bot)
+
+
+def setup(bot: commands.Bot) -> None:
+    bot.add_cog(KojiiCharacterCog(bot))
diff --git a/src/bots/kojii/KojiiGeneratorCog.py b/src/bots/kojii/KojiiGeneratorCog.py
@@ -1,18 +1,11 @@
 from discord.ext import commands
+from cogs.CharacterGeneratorCog import CharacterGeneratorCog
 
-from cogs.GeneratorCog import GeneratorCog, LoraInput
 
-
-class KojiiGeneratorCog(GeneratorCog):
+class KojiiCharacterGeneratorCog(CharacterGeneratorCog):
     def __init__(self, bot: commands.bot) -> None:
-        lora = LoraInput(
-            lora_id="65642f6a730b5e00f6f243d6",
-            lora_strength=0.65,
-            lora_trigger="kojii",
-            require_lora_trigger=True,
-        )
-        super().__init__(bot, lora)
+        super().__init__(bot)
 
 
 def setup(bot: commands.Bot) -> None:
-    bot.add_cog(KojiiGeneratorCog(bot))
+    bot.add_cog(KojiiCharacterGeneratorCog(bot))