git commit -am 'Update space' && git push

app.py

@@ -12,7 +12,11 @@ if torch.cuda.is_available():
     pipe.enable_xformers_memory_efficient_attention()
     pipe = pipe.to(device)
 else: 
-    pipe = DiffusionPipeline.from_pretrained("stabilityai/sdxl-turbo", use_safetensors=True)
+    # pipe = DiffusionPipeline.from_pretrained("stabilityai/sdxl-turbo", use_safetensors=True)
+    model_id1 = "dreamlike-art/dreamlike-diffusion-1.0"
+    model_id2 = "stabilityai/stable-diffusion-xl-base-1.0"
+    model_id3 = "stabilityai/stable-diffusion-2"
+    pipe = StableDiffusionPipeline.from_pretrained(model_id1, torch_dtype=torch.float16, use_safetensors=True)
     pipe = pipe.to(device)
 
 MAX_SEED = np.iinfo(np.int32).max