diff --git a/cogs/entertainment/imagine.py b/cogs/entertainment/imagine.py
new file mode 100644
index 0000000..9bcec15
--- /dev/null
+++ b/cogs/entertainment/imagine.py
@@ -0,0 +1,12 @@
+import discord
+from discord.ext import commands
+
+from diffusers import DiffusionPipeline
+import torch
+
+
+class Imagine(commands.Cog):
+
+    def __init__(self, bot: commands.Bot):
+        self.pipe = DiffusionPipeline.from_pretrained("hakurei/waifu-diffusion", torch_dtype=torch.float16)
+        self.pipe.to("cuda")