Srikumar26
commited on
Commit
•
d890fd9
1
Parent(s):
8ba122d
Update app.py
Browse files
app.py
CHANGED
@@ -2,8 +2,7 @@ import torch
|
|
2 |
from diffusers import StableDiffusionPipeline, DDIMScheduler
|
3 |
import gradio as gr
|
4 |
|
5 |
-
pipe = StableDiffusionPipeline.from_pretrained("MVRL/GeoSynth")
|
6 |
-
pipe.enable_xformers_memory_efficient_attention()
|
7 |
# scheduler = DDIMScheduler.from_pretrained("stabilityai/stable-diffusion-2-1-base")
|
8 |
# pipe.scheduler = scheduler
|
9 |
|
|
|
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 |
|