Vivawaves commited on
Commit
9369b80
1 Parent(s): b0d663e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
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-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()
@@ -68,7 +68,7 @@ def randomize_seed_fn(seed: int, randomize_seed: bool) -> int:
68
 
69
  def generate(
70
  prompt: str,
71
- negative_prompt: str = "",
72
  seed: int = 0,
73
  width: int = 1024,
74
  height: int = 1024,
@@ -141,7 +141,7 @@ with gr.Blocks(css="style.css") as demo:
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(
 
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-base", 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()
 
68
 
69
  def generate(
70
  prompt: str,
71
+ negative_prompt: str = "bad anatomy, blurry, fuzzy, extra arms, extra fingers, poorly drawn hands, disfigured, tiling, deformed, mutated, drawing, imperfections",
72
  seed: int = 0,
73
  width: int = 1024,
74
  height: int = 1024,
 
141
  label="What I do NOT want",
142
  max_lines=1,
143
  placeholder="Uncheck seed to iterate and finetune.",
144
+ value="Example: Text"
145
  )
146
 
147
  seed = gr.Slider(