Test new fix

app.py

@@ -3,10 +3,10 @@ import torch
 from torch import autocast
 from diffusers import StableDiffusionPipeline
 
-model_id = "hakurei/waifu-diffusion"
-pipe = StableDiffusionPipeline.from_pretrained(model_id)
+#model_id = "hakurei/waifu-diffusion"
+pipe = StableDiffusionPipeline.from_pretrained("hakurei/waifu-diffusion", torch_type=torch.float16, revision="fp16")
 pipe = pipe.to("cuda")
-torch.backends.cudnn.benchmark = True
+#torch.backends.cudnn.benchmark = True
 num_samples = 2
 
 def infer(prompt):