wyysf commited on
Commit
096ef9e
β€’
1 Parent(s): b5c231b

Update gradio_app.py

Browse files
Files changed (1) hide show
  1. gradio_app.py +4 -4
gradio_app.py CHANGED
@@ -244,8 +244,8 @@ if __name__=="__main__":
244
  })
245
  mvimg_model_config_list = [
246
  "Era3D",
247
- # "CRM",
248
- # "ImageDream"
249
  ]
250
  if "Era3D" in mvimg_model_config_list:
251
  # cfg = load_config("apps/third_party/Era3D/configs/test_unclip-512-6view.yaml")
@@ -256,7 +256,7 @@ if __name__=="__main__":
256
  torch_dtype=torch.float16
257
  )
258
  # enable xformers
259
- era3d_pipeline.unet.enable_xformers_memory_efficient_attention()
260
  era3d_pipeline.to(device)
261
  elif "CRM" in mvimg_model_config_list:
262
  stage1_config = OmegaConf.load(f"apps/third_party/CRM/configs/nf7_v3_SNR_rd_size_stroke.yaml").config
@@ -321,7 +321,7 @@ if __name__=="__main__":
321
  gr.Markdown('''Try a different <b>seed and MV Model</b> for better results. Good Luck :)''')
322
  with gr.Row():
323
  seed = gr.Number(0, label='Seed', show_label=True)
324
- mvimg_model = gr.Dropdown(value="Era3D", label="MV Image Model", choices=list(mvimg_model_config_list))
325
  more = gr.CheckboxGroup(["Remesh", "Symmetry(TBD)"], label="More", show_label=False)
326
  with gr.Row():
327
  # input prompt
 
244
  })
245
  mvimg_model_config_list = [
246
  "Era3D",
247
+ "CRM",
248
+ "ImageDream"
249
  ]
250
  if "Era3D" in mvimg_model_config_list:
251
  # cfg = load_config("apps/third_party/Era3D/configs/test_unclip-512-6view.yaml")
 
256
  torch_dtype=torch.float16
257
  )
258
  # enable xformers
259
+ # era3d_pipeline.unet.enable_xformers_memory_efficient_attention()
260
  era3d_pipeline.to(device)
261
  elif "CRM" in mvimg_model_config_list:
262
  stage1_config = OmegaConf.load(f"apps/third_party/CRM/configs/nf7_v3_SNR_rd_size_stroke.yaml").config
 
321
  gr.Markdown('''Try a different <b>seed and MV Model</b> for better results. Good Luck :)''')
322
  with gr.Row():
323
  seed = gr.Number(0, label='Seed', show_label=True)
324
+ mvimg_model = gr.Dropdown(value="CRM", label="MV Image Model", choices=list(mvimg_model_config_list))
325
  more = gr.CheckboxGroup(["Remesh", "Symmetry(TBD)"], label="More", show_label=False)
326
  with gr.Row():
327
  # input prompt