JoPmt commited on
Commit
6abe5da
1 Parent(s): c3daa7e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -74,7 +74,7 @@ def plex(prompt,stips,nut,wit,het,gaul,progress=gr.Progress(track_tqdm=True)):
74
  else:
75
  nm=nut
76
  generator = torch.Generator(device="cpu").manual_seed(nm)
77
- image = pipe(prompt=[prompt]*2,num_inference_steps=2,width=512,height=512,guidance_scale=0.0,generator=generator)
78
  for a, imze in enumerate(image["images"]):
79
  apol.append(imze)
80
  imze.save('./tmpo/'+fnamo+'_'+str(a)+'.png', 'PNG')
@@ -93,7 +93,7 @@ with gr.Blocks(theme=random.choice([gr.themes.Monochrome(),gr.themes.Base.from_h
93
  gaul=gr.Textbox(visible=False)
94
  btn=gr.Button("GENERATE")
95
  with gr.Accordion("Advanced Settings", open=False):
96
- inyt=gr.Slider(label="Num inference steps",minimum=1,step=1,maximum=30,value=20)
97
  indt=gr.Slider(label="Manual seed (leave 0 for random)",minimum=0,step=32,maximum=2147483616,value=0)
98
  inwt=gr.Slider(label="Width",minimum=256,step=32,maximum=768,value=512)
99
  inht=gr.Slider(label="Height",minimum=256,step=32,maximum=768,value=512)
 
74
  else:
75
  nm=nut
76
  generator = torch.Generator(device="cpu").manual_seed(nm)
77
+ image = pipe(prompt=[prompt]*2,num_inference_steps=stips,width=512,height=512,guidance_scale=0.0,generator=generator)
78
  for a, imze in enumerate(image["images"]):
79
  apol.append(imze)
80
  imze.save('./tmpo/'+fnamo+'_'+str(a)+'.png', 'PNG')
 
93
  gaul=gr.Textbox(visible=False)
94
  btn=gr.Button("GENERATE")
95
  with gr.Accordion("Advanced Settings", open=False):
96
+ inyt=gr.Slider(label="Num inference steps",minimum=2,step=1,maximum=10,value=4)
97
  indt=gr.Slider(label="Manual seed (leave 0 for random)",minimum=0,step=32,maximum=2147483616,value=0)
98
  inwt=gr.Slider(label="Width",minimum=256,step=32,maximum=768,value=512)
99
  inht=gr.Slider(label="Height",minimum=256,step=32,maximum=768,value=512)