wyysf commited on
Commit
dd81b16
1 Parent(s): e0da7fb

Update gradio_app.py

Browse files
Files changed (1) hide show
  1. gradio_app.py +1 -2
gradio_app.py CHANGED
@@ -254,8 +254,7 @@ if __name__=="__main__":
254
  # schema = OmegaConf.structured(TestConfig)
255
  # cfg = OmegaConf.merge(schema, cfg)
256
  era3d_pipeline = StableUnCLIPImg2ImgPipeline.from_pretrained(
257
- 'pengHTYX/MacLab-Era3D-512-6view',
258
- torch_dtype=torch.float16
259
  )
260
  # enable xformers
261
  # era3d_pipeline.unet.enable_xformers_memory_efficient_attention()
 
254
  # schema = OmegaConf.structured(TestConfig)
255
  # cfg = OmegaConf.merge(schema, cfg)
256
  era3d_pipeline = StableUnCLIPImg2ImgPipeline.from_pretrained(
257
+ 'pengHTYX/MacLab-Era3D-512-6view'
 
258
  )
259
  # enable xformers
260
  # era3d_pipeline.unet.enable_xformers_memory_efficient_attention()