Update app.py
Browse files
app.py
CHANGED
|
@@ -2,7 +2,8 @@ import torch
|
|
| 2 |
from diffusers import StableDiffusionPipeline, DDIMScheduler
|
| 3 |
import gradio as gr
|
| 4 |
|
| 5 |
-
pipe = StableDiffusionPipeline.from_pretrained("MVRL/GeoSynth")
|
|
|
|
| 6 |
# scheduler = DDIMScheduler.from_pretrained("stabilityai/stable-diffusion-2-1-base")
|
| 7 |
# pipe.scheduler = scheduler
|
| 8 |
|
|
|
|
| 2 |
from diffusers import StableDiffusionPipeline, DDIMScheduler
|
| 3 |
import gradio as gr
|
| 4 |
|
| 5 |
+
pipe = StableDiffusionPipeline.from_pretrained("MVRL/GeoSynth").cuda()
|
| 6 |
+
pipe.enable_xformers_memory_efficient_attention()
|
| 7 |
# scheduler = DDIMScheduler.from_pretrained("stabilityai/stable-diffusion-2-1-base")
|
| 8 |
# pipe.scheduler = scheduler
|
| 9 |
|