Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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=
|
183 |
)
|
184 |
prior_num_inference_steps = gr.Slider(
|
185 |
label="Prior Inference Steps",
|
186 |
minimum=30,
|
187 |
-
maximum=
|
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 |
)
|