fix

app.py

@@ -110,13 +110,11 @@ def load_pipeline(
     # Optionally set memory layout.
     if use_channels_last:
         print("Setting memory layout.")
-        if pipeline_to_benchmark not in ["Würstchen (T2I)", "Kandinsky 2.2 (T2I)"]:
+        if pipeline_to_benchmark != "Würstchen (T2I)":
             pipeline.unet.to(memory_format=torch.channels_last)
         elif pipeline_to_benchmark == "Würstchen (T2I)":
             pipeline.prior_prior.to(memory_format=torch.channels_last)
             pipeline.decoder.to(memory_format=torch.channels_last)
-        elif pipeline_to_benchmark == "Kandinsky 2.2 (T2I)":
-            pipeline.unet.to(memory_format=torch.channels_last)
 
         if hasattr(pipeline, "controlnet"):
             pipeline.controlnet.to(memory_format=torch.channels_last)
@@ -126,7 +124,7 @@ def load_pipeline(
     # Optional torch compilation.
     if do_torch_compile:
         print("Compiling pipeline.")
-        if pipeline_to_benchmark not in ["Würstchen (T2I)", "Kandinsky 2.2 (T2I)"]:
+        if pipeline_to_benchmark != "Würstchen (T2I)":
             pipeline.unet = torch.compile(
                 pipeline.unet, mode="reduce-overhead", fullgraph=True
             )
@@ -137,10 +135,6 @@ def load_pipeline(
             pipeline.decoder = torch.compile(
                 pipeline.decoder, mode="reduce-overhead", fullgraph=True
             )
-        elif pipeline_to_benchmark == "Kandinsky 2.2 (T2I)":
-            pipeline.unet = torch.compile(
-                pipeline.unet, mode="reduce-overhead", fullgraph=True
-            )
 
         if hasattr(pipeline, "controlnet"):
             pipeline.controlnet = torch.compile(