LouisLi commited on
Commit
c070b97
·
verified ·
1 Parent(s): 713fc08

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -118,7 +118,7 @@ if cuda_path:
118
  else:
119
  print("CUDA installation not found")
120
 
121
- config_path = 'configs/instant-mesh-large.yaml'
122
  config = OmegaConf.load(config_path)
123
  config_name = os.path.basename(config_path).replace('.yaml', '')
124
  model_config = config.model_config
@@ -148,7 +148,7 @@ pipeline = pipeline.to(device)
148
 
149
  # load reconstruction model
150
  print('Loading reconstruction model ...')
151
- model_ckpt_path = hf_hub_download(repo_id="TencentARC/InstantMesh", filename="instant_mesh_large.ckpt", repo_type="model")
152
  model0 = instantiate_from_config(model_config)
153
  state_dict = torch.load(model_ckpt_path, map_location='cpu')['state_dict']
154
  state_dict = {k[14:]: v for k, v in state_dict.items() if k.startswith('lrm_generator.') and 'source_camera' not in k}
 
118
  else:
119
  print("CUDA installation not found")
120
 
121
+ config_path = 'configs/instant-nerf-base.yaml'
122
  config = OmegaConf.load(config_path)
123
  config_name = os.path.basename(config_path).replace('.yaml', '')
124
  model_config = config.model_config
 
148
 
149
  # load reconstruction model
150
  print('Loading reconstruction model ...')
151
+ model_ckpt_path = hf_hub_download(repo_id="TencentARC/InstantMesh", filename="instant_nerf_base.ckpt", repo_type="model")
152
  model0 = instantiate_from_config(model_config)
153
  state_dict = torch.load(model_ckpt_path, map_location='cpu')['state_dict']
154
  state_dict = {k[14:]: v for k, v in state_dict.items() if k.startswith('lrm_generator.') and 'source_camera' not in k}