pengHTYX commited on
Commit
2784a94
1 Parent(s): 9a8311b
Files changed (1) hide show
  1. app.py +6 -14
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.Accordion('Advanced options', open=True):
370
- with gr.Row():
371
- with gr.Column():
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
- with gr.Column():
 
379
  output_processing = gr.CheckboxGroup(
380
- ['Write Results'], label='write the results in ./outputs folder', value=['Write Results']
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():