wyysf commited on
Commit
e0da7fb
β€’
1 Parent(s): 1656878

Update gradio_app.py

Browse files
Files changed (1) hide show
  1. 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
- elif "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,7 +272,7 @@ if __name__=="__main__":
272
  device=device,
273
  dtype=torch.float16
274
  )
275
- elif "ImageDream" in mvimg_model_config_list:
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,