Update app.py

app.py

@@ -13,7 +13,7 @@ import gradio as gr
 import numpy as np
 from PIL import Image
 import torch
-import torch._dynamo
+#import torch._dynamo
 #import diffusers
 from diffusers import AutoencoderKL, StableDiffusionXLPipeline
 from diffusers import EulerAncestralDiscreteScheduler
@@ -198,7 +198,7 @@ def load_and_prepare_model():
     #pipe.unet = pipe.unet.to(memory_format=torch.contiguous_format)
     
         #Some typical diffusers pipeline optimizations
-    pipe.unet.to(memory_format=torch.channels_last) #Unsupported by hidet, but does not seem to make a difference if disabled.
+    #pipe.unet.to(memory_format=torch.channels_last) #Unsupported by hidet, but does not seem to make a difference if disabled.
     #pipe.enable_vae_tiling()
     #pipe.enable_xformers_memory_efficient_attention()
 
@@ -289,7 +289,7 @@ def generate_30(
     lora_scale: float = 0.5,
     progress=gr.Progress(track_tqdm=True)  # Add progress as a keyword argument
 ):
-    torch.set_default_device('cuda')
+    #torch.set_default_device('cuda')
     #pipe.set_adapters(["skin"], adapter_weights=[lora_scale])
     seed = int(randomize_seed_fn())
     generator = torch.Generator(device='cuda').manual_seed(seed)