Spaces:
Sleeping
Sleeping
FrozenBurning
commited on
Commit
•
e4b9c27
1
Parent(s):
2c17c74
Update app.py
Browse files- app.py +1 -1
- 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,
|
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)
|
|