aljaz/generative-models
1
0
Fork 0
mirror of https://github.com/Stability-AI/generative-models.git synced 2026-01-11 01:24:25 +01:00
Code Issues Packages Projects Releases Wiki Activity

SV4D: reduce the memory consumption and speed up

Browse Source
This commit is contained in:
ymxie97
2024-08-02 04:59:37 +00:00
parent e0596f1aca
commit 854bd4f0df
3 changed files with 253 additions and 114 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
scripts/sampling/configs/sv4d.yaml
View File

@@ -1,6 +1,7 @@
N_TIME: 5
N_VIEW: 8
N_FRAMES: 40
ENCODE_N_A_TIME: 8
model:
target: sgm.models.diffusion.DiffusionEngine
@@ -67,6 +68,7 @@ model:
is_ae: True
n_cond_frames: ${N_FRAMES}
n_copies: 1
en_and_decode_n_samples_a_time: ${ENCODE_N_A_TIME}
encoder_config:
target: sgm.models.autoencoder.AutoencoderKLModeOnly
params:
@@ -131,6 +133,7 @@ model:
is_ae: True
n_cond_frames: ${N_VIEW}
n_copies: 1
en_and_decode_n_samples_a_time: ${ENCODE_N_A_TIME}
sigma_sampler_config:
target: sgm.modules.diffusionmodules.sigma_sampling.ZeroSampler
@@ -141,6 +144,7 @@ model:
is_ae: True
n_cond_frames: ${N_TIME}
n_copies: 1
en_and_decode_n_samples_a_time: ${ENCODE_N_A_TIME}
encoder_config:
target: sgm.models.autoencoder.AutoencoderKLModeOnly
params: