Update app.py

app.py

@@ -25,7 +25,7 @@ from illusion_style import css
 BASE_MODEL = "SG161222/Realistic_Vision_V5.1_noVAE"
 
 
-device='cuda'
+device='cpu'
 
 # Initialize both pipelines
 vae = AutoencoderKL.from_pretrained("stabilityai/sd-vae-ft-mse", torch_dtype=torch.float16)
@@ -37,7 +37,7 @@ main_pipe = StableDiffusionControlNetPipeline.from_pretrained(
     vae=vae,
     safety_checker=None,
     torch_dtype=torch.float16,
-).to("cuda")
+).to(device)
 
 #main_pipe.unet = torch.compile(main_pipe.unet, mode="reduce-overhead", fullgraph=True)
 #main_pipe.unet.to(memory_format=torch.channels_last)
@@ -141,7 +141,7 @@ def inference(
 
     main_pipe.scheduler = SAMPLER_MAP[sampler](main_pipe.scheduler.config)
     my_seed = random.randint(0, 2**32 - 1) if seed == -1 else seed
-    generator = torch.Generator(device="cuda").manual_seed(my_seed)
+    generator = torch.Generator(device=device).manual_seed(my_seed)
     
     out = main_pipe(
         prompt=prompt,