Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -118,7 +118,7 @@ if cuda_path:
|
|
118 |
else:
|
119 |
print("CUDA installation not found")
|
120 |
|
121 |
-
config_path = 'configs/instant-
|
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="
|
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}
|