Spaces:
Runtime error
Runtime error
pengHTYX
commited on
Commit
•
2784a94
1
Parent(s):
9a8311b
'update'
Browse files
app.py
CHANGED
@@ -193,15 +193,6 @@ def run_pipeline(pipeline, cfg, single_image, guidance_scale, steps, seed, crop_
|
|
193 |
imgs_in = imgs_in.to(device=f'cuda:{_GPU_ID}', dtype=weight_dtype)
|
194 |
prompt_embeddings = prompt_embeddings.to(device=f'cuda:{_GPU_ID}', dtype=weight_dtype)
|
195 |
|
196 |
-
ic(pipeline.unet.device)
|
197 |
-
ic(pipeline.unet.dtype)
|
198 |
-
|
199 |
-
ic(imgs_in.device)
|
200 |
-
ic(imgs_in.dtype)
|
201 |
-
|
202 |
-
ic(prompt_embeddings.device)
|
203 |
-
ic(prompt_embeddings.dtype)
|
204 |
-
|
205 |
out = pipeline(
|
206 |
imgs_in,
|
207 |
None,
|
@@ -366,18 +357,19 @@ def run_demo():
|
|
366 |
examples_per_page=30,
|
367 |
)
|
368 |
with gr.Column(scale=1):
|
369 |
-
with gr.
|
370 |
-
with gr.
|
371 |
-
with gr.
|
372 |
input_processing = gr.CheckboxGroup(
|
373 |
['Background Removal'],
|
374 |
label='Input Image Preprocessing',
|
375 |
value=['Background Removal'],
|
376 |
info='untick this, if masked image with alpha channel',
|
377 |
)
|
378 |
-
|
|
|
379 |
output_processing = gr.CheckboxGroup(
|
380 |
-
['Write Results'], label='write the results in
|
381 |
)
|
382 |
with gr.Row():
|
383 |
with gr.Column():
|
|
|
193 |
imgs_in = imgs_in.to(device=f'cuda:{_GPU_ID}', dtype=weight_dtype)
|
194 |
prompt_embeddings = prompt_embeddings.to(device=f'cuda:{_GPU_ID}', dtype=weight_dtype)
|
195 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
196 |
out = pipeline(
|
197 |
imgs_in,
|
198 |
None,
|
|
|
357 |
examples_per_page=30,
|
358 |
)
|
359 |
with gr.Column(scale=1):
|
360 |
+
with gr.Row():
|
361 |
+
with gr.Column():
|
362 |
+
with gr.Accordion('Advanced options', open=True):
|
363 |
input_processing = gr.CheckboxGroup(
|
364 |
['Background Removal'],
|
365 |
label='Input Image Preprocessing',
|
366 |
value=['Background Removal'],
|
367 |
info='untick this, if masked image with alpha channel',
|
368 |
)
|
369 |
+
with gr.Column():
|
370 |
+
with gr.Accordion('Advanced options', open=False):
|
371 |
output_processing = gr.CheckboxGroup(
|
372 |
+
['Write Results'], label='write the results in mv_res folder', value=['Write Results']
|
373 |
)
|
374 |
with gr.Row():
|
375 |
with gr.Column():
|