Update app.py

app.py

@@ -81,7 +81,7 @@ neg_prompt_2 = " 'non-photorealistic':1.5, 'unrealistic skin','unattractive face
 
 device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
 
-upscaler = UpscaleWithModel.from_pretrained("Kim2091/ClearRealityV1").to(torch.device("cuda:0"))
+upscaler = UpscaleWithModel.from_pretrained("Kim2091/ClearRealityV1").to(torch.device("cpu"))
 
 def load_and_prepare_model():
     vaeXL = AutoencoderKL.from_pretrained("stabilityai/sdxl-vae", safety_checker=None, use_safetensors=False) #.to(device=device, dtype=torch.bfloat16)
@@ -228,8 +228,8 @@ def generate(
     if segment==8:
         final_latents = current_latents
         final_latents = final_latents / pipe.vae.config.scaling_factor
-        with torch.no_grad():
-            image = pipe.vae.decode(final_latents, return_dict=False)[0]
+        #with torch.no_grad():
+        image = pipe.vae.decode(final_latents, return_dict=False)[0]
         image = pipe.image_processor.postprocess(image, output_type="pil")[0]
         output_image_file = f"rv_L_{seed}.png"
         image.save(output_image_file)