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)
|