Update app.py

app.py

@@ -247,8 +247,8 @@ MAX_IMAGE_SIZE = 2048
 
 if torch.cuda.is_available():
     # vae = AutoencoderKL.from_pretrained("madebyollin/sdxl-vae-fp16-fix", torch_dtype=torch.float16)
-    pipe = StableDiffusionXLPipeline.from_pretrained("John6666/wai-nsfw-illustrious-sdxl-v140-sdxl")
-    pipe.scheduler = EulerAncestralDiscreteScheduler.from_config(pipe.scheduler.config)
+    pipe = StableDiffusionXLPipeline.from_pretrained("John6666/wai-nsfw-illustrious-sdxl-v140-sdxl",torch_dtype=torch.float16)
+    # pipe.scheduler = EulerAncestralDiscreteScheduler.from_config(pipe.scheduler.config)
     pipe.to("cuda")
 
 def randomize_seed_fn(seed: int, randomize_seed: bool) -> int: