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
|