Srikumar26 commited on
Commit
22ef354
·
verified ·
1 Parent(s): ec04562

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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)