FrozenBurning commited on
Commit
ef4bc82
1 Parent(s): 692682d

fix app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -141,7 +141,7 @@ def process(input_image, input_num_steps=25, input_seed=42, input_cfg=6.0):
141
  # exporting GLB mesh
142
  denoise_param_path = os.path.join(config.output_dir, 'denoised.pt')
143
  primx_ckpt_weight = torch.load(denoise_param_path, map_location='cpu')['model_state_dict']
144
- model_primx.load_state_dict(ckpt_weight)
145
  model_primx.to(device)
146
  model_primx.eval()
147
  with torch.no_grad():
 
141
  # exporting GLB mesh
142
  denoise_param_path = os.path.join(config.output_dir, 'denoised.pt')
143
  primx_ckpt_weight = torch.load(denoise_param_path, map_location='cpu')['model_state_dict']
144
+ model_primx.load_state_dict(primx_ckpt_weight)
145
  model_primx.to(device)
146
  model_primx.eval()
147
  with torch.no_grad():