Update gradio_app.py
Browse files- gradio_app.py +2 -2
gradio_app.py
CHANGED
|
@@ -260,7 +260,7 @@ if __name__=="__main__":
|
|
| 260 |
# enable xformers
|
| 261 |
# era3d_pipeline.unet.enable_xformers_memory_efficient_attention()
|
| 262 |
# era3d_pipeline.to(device)
|
| 263 |
-
|
| 264 |
stage1_config = OmegaConf.load(f"apps/third_party/CRM/configs/nf7_v3_SNR_rd_size_stroke.yaml").config
|
| 265 |
stage1_sampler_config = stage1_config.sampler
|
| 266 |
stage1_model_config = stage1_config.models
|
|
@@ -272,7 +272,7 @@ if __name__=="__main__":
|
|
| 272 |
device=device,
|
| 273 |
dtype=torch.float16
|
| 274 |
)
|
| 275 |
-
|
| 276 |
imagedream_pipeline = MVDreamPipeline.from_pretrained(
|
| 277 |
"ashawkey/imagedream-ipmv-diffusers", # remote weights
|
| 278 |
torch_dtype=torch.float16,
|
|
|
|
| 260 |
# enable xformers
|
| 261 |
# era3d_pipeline.unet.enable_xformers_memory_efficient_attention()
|
| 262 |
# era3d_pipeline.to(device)
|
| 263 |
+
if "CRM" in mvimg_model_config_list:
|
| 264 |
stage1_config = OmegaConf.load(f"apps/third_party/CRM/configs/nf7_v3_SNR_rd_size_stroke.yaml").config
|
| 265 |
stage1_sampler_config = stage1_config.sampler
|
| 266 |
stage1_model_config = stage1_config.models
|
|
|
|
| 272 |
device=device,
|
| 273 |
dtype=torch.float16
|
| 274 |
)
|
| 275 |
+
if "ImageDream" in mvimg_model_config_list:
|
| 276 |
imagedream_pipeline = MVDreamPipeline.from_pretrained(
|
| 277 |
"ashawkey/imagedream-ipmv-diffusers", # remote weights
|
| 278 |
torch_dtype=torch.float16,
|