Default to jit_compile=True for SD (#2054)

Author: ianstenbit

keras_cv/models/stable_diffusion/stable_diffusion.py

@@ -52,7 +52,7 @@ def __init__(
         self,
         img_height=512,
         img_width=512,
-        jit_compile=False,
+        jit_compile=True,
     ):
         # UNet requires multiples of 2**7 = 128
         img_height = round(img_height / 128) * 128
@@ -396,7 +396,7 @@ def __init__(
         self,
         img_height=512,
         img_width=512,
-        jit_compile=False,
+        jit_compile=True,
     ):
         super().__init__(img_height, img_width, jit_compile)
         print(
@@ -482,7 +482,7 @@ def __init__(
         self,
         img_height=512,
         img_width=512,
-        jit_compile=False,
+        jit_compile=True,
     ):
         super().__init__(img_height, img_width, jit_compile)
         print(

keras_cv/models/stable_diffusion/stable_diffusion_test.py

@@ -100,6 +100,7 @@ def test_generate_image_rejects_noise_and_seed(self):
 
 @pytest.mark.extra_large
 class StableDiffusionMultiFrameworkTest(TestCase):
+    @pytest.mark.filterwarnings("ignore::UserWarning")  # Torch + jit_compile
     def test_end_to_end(self):
         prompt = "a caterpillar smoking a hookah while sitting on a mushroom"
         stablediff = StableDiffusion(128, 128)