Huiwenshi commited on
Commit
5e028db
·
verified ·
1 Parent(s): 11a7ef1

Upload app_hg.py with huggingface_hub

Browse files
Files changed (1) hide show
  1. app_hg.py +4 -4
app_hg.py CHANGED
@@ -72,7 +72,7 @@ parser.add_argument("--use_lite", default=False, action="store_true")
72
  parser.add_argument("--mv23d_cfg_path", default="./svrm/configs/svrm.yaml", type=str)
73
  parser.add_argument("--mv23d_ckt_path", default="weights/svrm/svrm.safetensors", type=str)
74
  parser.add_argument("--text2image_path", default="weights/hunyuanDiT", type=str)
75
- parser.add_argument("--save_memory", default=True)
76
  parser.add_argument("--device", default="cuda:0", type=str)
77
  args = parser.parse_args()
78
 
@@ -237,7 +237,7 @@ def stage_2_i2v(image, seed, step, save_folder):
237
  show_img = Image.fromarray(show_img)
238
  return views_img, cond_img, show_img
239
 
240
- @spaces.GPU(duration=120)
241
  def stage_3_v23(
242
  views_pil,
243
  cond_pil,
@@ -332,7 +332,7 @@ with gr.Blocks() as demo:
332
  with gr.Row():
333
  textgen_render = gr.Checkbox(label="Do Rendering", value=True, interactive=True)
334
  if BAKE_AVAILEBLE:
335
- textgen_bake = gr.Checkbox(label="Do Baking", value=True, interactive=True)
336
  else:
337
  textgen_bake = gr.Checkbox(label="Do Baking", value=False, interactive=False)
338
 
@@ -367,7 +367,7 @@ with gr.Blocks() as demo:
367
  imggen_removebg = gr.Checkbox(label="Remove Background", value=True, interactive=True)
368
  imggen_render = gr.Checkbox(label="Do Rendering", value=True, interactive=True)
369
  if BAKE_AVAILEBLE:
370
- imggen_bake = gr.Checkbox(label="Do Baking", value=True, interactive=True)
371
  else:
372
  imggen_bake = gr.Checkbox(label="Do Baking", value=False, interactive=False)
373
 
 
72
  parser.add_argument("--mv23d_cfg_path", default="./svrm/configs/svrm.yaml", type=str)
73
  parser.add_argument("--mv23d_ckt_path", default="weights/svrm/svrm.safetensors", type=str)
74
  parser.add_argument("--text2image_path", default="weights/hunyuanDiT", type=str)
75
+ parser.add_argument("--save_memory", default=False)
76
  parser.add_argument("--device", default="cuda:0", type=str)
77
  args = parser.parse_args()
78
 
 
237
  show_img = Image.fromarray(show_img)
238
  return views_img, cond_img, show_img
239
 
240
+ @spaces.GPU(duration=70)
241
  def stage_3_v23(
242
  views_pil,
243
  cond_pil,
 
332
  with gr.Row():
333
  textgen_render = gr.Checkbox(label="Do Rendering", value=True, interactive=True)
334
  if BAKE_AVAILEBLE:
335
+ textgen_bake = gr.Checkbox(label="Do Baking", value=False, interactive=True)
336
  else:
337
  textgen_bake = gr.Checkbox(label="Do Baking", value=False, interactive=False)
338
 
 
367
  imggen_removebg = gr.Checkbox(label="Remove Background", value=True, interactive=True)
368
  imggen_render = gr.Checkbox(label="Do Rendering", value=True, interactive=True)
369
  if BAKE_AVAILEBLE:
370
+ imggen_bake = gr.Checkbox(label="Do Baking", value=False, interactive=True)
371
  else:
372
  imggen_bake = gr.Checkbox(label="Do Baking", value=False, interactive=False)
373