Update gradio_app.py
Browse files- 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 |
-
|
| 248 |
-
|
| 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="
|
| 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
|