Update app.py
Browse files
app.py
CHANGED
@@ -3,8 +3,8 @@ 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 |
|
9 |
def process(prompt, n_prompt, num_samples, image_resolution, ddim_steps, scale, seed, eta):
|
10 |
generator = torch.manual_seed(seed)
|
|
|
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 |
|
9 |
def process(prompt, n_prompt, num_samples, image_resolution, ddim_steps, scale, seed, eta):
|
10 |
generator = torch.manual_seed(seed)
|