Vivawaves commited on
Commit
b0d663e
1 Parent(s): a0f560f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -4
app.py CHANGED
@@ -29,7 +29,7 @@ PREVIEW_IMAGES = True
29
  dtype = torch.float16
30
  device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
31
  if torch.cuda.is_available():
32
- prior_pipeline = WuerstchenPriorPipeline.from_pretrained("warp-ai/wuerstchen-prior", torch_dtype=dtype)
33
  decoder_pipeline = WuerstchenDecoderPipeline.from_pretrained("warp-ai/wuerstchen", torch_dtype=dtype)
34
  if ENABLE_CPU_OFFLOAD:
35
  prior_pipeline.enable_model_cpu_offload()
@@ -140,7 +140,8 @@ with gr.Blocks(css="style.css") as demo:
140
  negative_prompt = gr.Text(
141
  label="What I do NOT want",
142
  max_lines=1,
143
- placeholder="Uncheck seed to finetune.",
 
144
  )
145
 
146
  seed = gr.Slider(
@@ -179,12 +180,12 @@ with gr.Blocks(css="style.css") as demo:
179
  minimum=0,
180
  maximum=20,
181
  step=0.1,
182
- value=4.0,
183
  )
184
  prior_num_inference_steps = gr.Slider(
185
  label="Prior Inference Steps",
186
  minimum=30,
187
- maximum=30,
188
  step=1,
189
  value=30,
190
  )
 
29
  dtype = torch.float16
30
  device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
31
  if torch.cuda.is_available():
32
+ prior_pipeline = WuerstchenPriorPipeline.from_pretrained("warp-ai/wuerstchen-prior-model-interpolated", torch_dtype=dtype)
33
  decoder_pipeline = WuerstchenDecoderPipeline.from_pretrained("warp-ai/wuerstchen", torch_dtype=dtype)
34
  if ENABLE_CPU_OFFLOAD:
35
  prior_pipeline.enable_model_cpu_offload()
 
140
  negative_prompt = gr.Text(
141
  label="What I do NOT want",
142
  max_lines=1,
143
+ placeholder="Uncheck seed to iterate and finetune.",
144
+ value="lowres, text, error, cropped, worst quality, low quality, jpeg artifacts, ugly, duplicate, morbid, mutilated, out of frame, extra fingers, mutated hands, poorly drawn hands, poorly drawn face, mutation, deformed, blurry, dehydrated, bad anatomy, bad proportions, extra limbs, cloned face, disfigured, gross proportions, malformed limbs, missing arms, missing legs, extra arms, extra legs, fused fingers, too many fingers, long neck, username, watermark, signature"
145
  )
146
 
147
  seed = gr.Slider(
 
180
  minimum=0,
181
  maximum=20,
182
  step=0.1,
183
+ value=17.0,
184
  )
185
  prior_num_inference_steps = gr.Slider(
186
  label="Prior Inference Steps",
187
  minimum=30,
188
+ maximum=60,
189
  step=1,
190
  value=30,
191
  )