Update app.py

app.py

@@ -17,7 +17,7 @@ from diffusers import StableDiffusionXLPipeline, LCMScheduler
 
 # from scheduling_tcd import TCDScheduler
 
-torch.backends.cuda.matmul.allow_tf32 = True
+# torch.backends.cuda.matmul.allow_tf32 = True
 
 class timer:
     def __init__(self, method_name="timed process"):
@@ -35,7 +35,7 @@ if not path.exists(cache_path):
     os.makedirs(cache_path, exist_ok=True)
 
 pipe = StableDiffusionXLPipeline.from_pretrained("stabilityai/stable-diffusion-xl-base-1.0", torch_dtype=torch.bfloat16)
-pipe.to(device="cuda", dtype=torch.bfloat16)
+pipe.to(device="cpu", dtype=torch.bfloat16)
 unet_state = load_file(hf_hub_download("ByteDance/Hyper-SD", "Hyper-SDXL-1step-Unet.safetensors"), device="cuda")
 pipe.unet.load_state_dict(unet_state)
 pipe.scheduler = LCMScheduler.from_config(pipe.scheduler.config, timestep_spacing ="trailing")