Add SD v1.5

app.py

@@ -8,7 +8,9 @@ import torch
 from torch import autocast
 
 from transformers import pipeline, set_seed
-from diffusers import StableDiffusionPipeline
+from diffusers import DiffusionPipeline, StableDiffusionPipeline
+
+
 
 # GPT2
 generator = pipeline('text-generation', model='gpt2')
@@ -19,4 +21,7 @@ generator("Hello world, I'm vizard,", max_length=50, num_return_sequences=3)
 # SD v1.4
 pipe = StableDiffusionPipeline.from_pretrained(model_id, use_auth_token=True, revision="fp16", torch_dtype=torch.float16)
 pipe = pipe.to(device)
-torch.backends.cudnn.benchmark = True
+torch.backends.cudnn.benchmark = True
+
+# SD v1.5
+pipeline = DiffusionPipeline.from_pretrained("runwayml/stable-diffusion-v1-5")