Update app.py

app.py

@@ -8,11 +8,11 @@ move_cache()
 
 model_path = "slymnyldrm/slymnyldrm3"
 
-pipe = StableDiffusionPipeline.from_pretrained(model_path, use_safetensors=True, safety_checker=None, torch_dtype=torch.float16).to("cuda")
+pipe = StableDiffusionPipeline.from_pretrained(model_path, use_safetensors=True, safety_checker=None, torch_dtype=torch.float16)
 pipe.scheduler = DDIMScheduler.from_config(pipe.scheduler.config)
 pipe.enable_xformers_memory_efficient_attention()
 
-g_cuda = torch.Generator(device='cuda')
+g_cuda = torch.Generator()
 
 def inference(prompt, negative_prompt, num_samples, height=512, width=512, num_inference_steps=50, guidance_scale=7.5, g_seed=52362):
     with torch.autocast("cuda"), torch.inference_mode():