Manjushri commited on
Commit
f7cf7f1
·
verified ·
1 Parent(s): b08bb51

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -3
app.py CHANGED
@@ -10,7 +10,7 @@ device = 'cuda' if torch.cuda.is_available() else 'cpu'
10
  torch.cuda.max_memory_allocated(device=device)
11
  torch.cuda.empty_cache()
12
 
13
- def genie (Model, Prompt, negative_prompt, height, width, scale, steps, seed, refine, high_noise_frac):
14
  generator = np.random.seed(0) if seed == 0 else torch.manual_seed(seed)
15
 
16
  if Model == "PhotoReal":
@@ -65,8 +65,7 @@ gr.Interface(fn=genie, inputs=[gr.Radio(['PhotoReal', 'Animagine XL 3.0',], valu
65
  gr.Slider(1, maximum=15, value=5, step=.25, label='Guidance Scale'),
66
  gr.Slider(5, maximum=100, value=50, step=5, label='Number of Iterations'),
67
  gr.Slider(minimum=0, step=1, maximum=9999999999999999, randomize=True, label='Seed: 0 is Random'),
68
- gr.Radio(["Yes", "No"], label='SDXL 1.0 Refiner: Use if the Image has too much Noise', value='No'),
69
- gr.Slider(minimum=.9, maximum=.99, value=.95, step=.01, label='Refiner Denoise Start %')],
70
  outputs=gr.Image(label='Generated Image'),
71
  title="Manju Dream Booth V2.1 with SDXL 1.0 Refiner - GPU",
72
  description="<br><br><b/>Warning: This Demo is capable of producing NSFW content.",
 
10
  torch.cuda.max_memory_allocated(device=device)
11
  torch.cuda.empty_cache()
12
 
13
+ def genie (Model, Prompt, negative_prompt, height, width, scale, steps, seed):
14
  generator = np.random.seed(0) if seed == 0 else torch.manual_seed(seed)
15
 
16
  if Model == "PhotoReal":
 
65
  gr.Slider(1, maximum=15, value=5, step=.25, label='Guidance Scale'),
66
  gr.Slider(5, maximum=100, value=50, step=5, label='Number of Iterations'),
67
  gr.Slider(minimum=0, step=1, maximum=9999999999999999, randomize=True, label='Seed: 0 is Random'),
68
+ ],
 
69
  outputs=gr.Image(label='Generated Image'),
70
  title="Manju Dream Booth V2.1 with SDXL 1.0 Refiner - GPU",
71
  description="<br><br><b/>Warning: This Demo is capable of producing NSFW content.",