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,
|