From 3dffd956feae72eb0e49a78618513739afa226cc Mon Sep 17 00:00:00 2001
From: terrance <terrance.3512@gmail.com>
Date: Fri, 3 Feb 2023 22:36:04 +0000
Subject: [PATCH] add warning if using non jit unet

---
 nerfstudio/generative/stable_diffusion.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/nerfstudio/generative/stable_diffusion.py b/nerfstudio/generative/stable_diffusion.py
index 1816da32d..8c4230f7e 100644
--- a/nerfstudio/generative/stable_diffusion.py
+++ b/nerfstudio/generative/stable_diffusion.py
@@ -109,6 +109,9 @@ class StableDiffusion(nn.Module):
             self.unet = TracedUNet()
             del pipe.unet
         else:
+            CONSOLE.print(
+                "Warning: Loading UNet without JIT acceleration. Run trace_stablediff.py in the scripts folder for a speedup!"
+            )
             self.unet = pipe.unet
             self.unet.to(memory_format=torch.channels_last)