FrozenBurning commited on
Commit
e4b9c27
1 Parent(s): 2c17c74

Update app.py

Browse files
Files changed (2) hide show
  1. app.py +1 -1
  2. assets/examples/firedragon.png +0 -0
app.py CHANGED
@@ -137,7 +137,7 @@ def process(input_cond, input_num_steps, input_seed=42, input_cfg=6.0):
137
  recon_feat_param[:, 1:, ...] = (recon_feat_param[:, 1:, ...] + 1) / 2.
138
  recon_feat_param = recon_feat_param.reshape(inf_bs, config.model.num_prims, -1)
139
  recon_param = torch.concat([recon_srt_param, recon_feat_param], dim=-1)
140
- visualize_video_primvolume(config.output_dir, batch, recon_param, 60, rm, device)
141
  prim_params = {'srt_param': recon_srt_param[0].detach().cpu(), 'feat_param': recon_feat_param[0].detach().cpu()}
142
  torch.save({'model_state_dict': prim_params}, "{}/denoised.pt".format(config.output_dir))
143
 
 
137
  recon_feat_param[:, 1:, ...] = (recon_feat_param[:, 1:, ...] + 1) / 2.
138
  recon_feat_param = recon_feat_param.reshape(inf_bs, config.model.num_prims, -1)
139
  recon_param = torch.concat([recon_srt_param, recon_feat_param], dim=-1)
140
+ visualize_video_primvolume(config.output_dir, batch, recon_param, 30, rm, device)
141
  prim_params = {'srt_param': recon_srt_param[0].detach().cpu(), 'feat_param': recon_feat_param[0].detach().cpu()}
142
  torch.save({'model_state_dict': prim_params}, "{}/denoised.pt".format(config.output_dir))
143
 
assets/examples/firedragon.png DELETED
Binary file (8.07 kB)