Path	Lines of Code
sat/app.py	1031
sat/arguments.py	251
sat/configs/tora/inference_sparse.yaml	13
sat/configs/tora/model/cogvideox_5b_tora.yaml	139
sat/configs/tora/model/cogvideox_5b_tora_i2v.yaml	145
sat/configs/tora/train_dense.yaml	63
sat/configs/tora/train_sparse.yaml	63
sat/data_video.py	372
sat/diffusion_video.py	324
sat/dit_video_concat.py	691
sat/sample_video.py	347
sat/sgm/__init__.py	3
sat/sgm/lr_scheduler.py	89
sat/sgm/models/__init__.py	1
sat/sgm/models/autoencoder.py	461
sat/sgm/modules/__init__.py	5
sat/sgm/modules/attention.py	440
sat/sgm/modules/autoencoding/__init__.py	1
sat/sgm/modules/autoencoding/losses/__init__.py	7
sat/sgm/modules/autoencoding/losses/discriminator_loss.py	236
sat/sgm/modules/autoencoding/losses/lpips.py	57
sat/sgm/modules/autoencoding/losses/video_loss.py	487
sat/sgm/modules/autoencoding/lpips/__init__.py	1
sat/sgm/modules/autoencoding/lpips/loss/__init__.py	1
sat/sgm/modules/autoencoding/lpips/loss/lpips.py	107
sat/sgm/modules/autoencoding/lpips/model/__init__.py	1
sat/sgm/modules/autoencoding/lpips/model/model.py	67
sat/sgm/modules/autoencoding/lpips/util.py	88
sat/sgm/modules/autoencoding/lpips/vqperceptual.py	12
sat/sgm/modules/autoencoding/magvit2_pytorch.py	1148
sat/sgm/modules/autoencoding/regularizers/__init__.py	24
sat/sgm/modules/autoencoding/regularizers/base.py	24
sat/sgm/modules/autoencoding/regularizers/finite_scalar_quantization.py	103
sat/sgm/modules/autoencoding/regularizers/lookup_free_quantization.py	154
sat/sgm/modules/autoencoding/regularizers/quantize.py	338
sat/sgm/modules/autoencoding/temporal_ae.py	277
sat/sgm/modules/autoencoding/vqvae/movq_dec_3d.py	379
sat/sgm/modules/autoencoding/vqvae/movq_dec_3d_dev.py	384
sat/sgm/modules/autoencoding/vqvae/movq_enc_3d.py	289
sat/sgm/modules/autoencoding/vqvae/movq_modules.py	285
sat/sgm/modules/autoencoding/vqvae/quantize.py	179
sat/sgm/modules/autoencoding/vqvae/vqvae_blocks.py	300
sat/sgm/modules/cp_enc_dec.py	101
sat/sgm/modules/diffusionmodules/__init__.py	6
sat/sgm/modules/diffusionmodules/denoiser.py	56
sat/sgm/modules/diffusionmodules/denoiser_scaling.py	46
sat/sgm/modules/diffusionmodules/denoiser_weighting.py	15
sat/sgm/modules/diffusionmodules/discretizer.py	102
sat/sgm/modules/diffusionmodules/guiders.py	65
sat/sgm/modules/diffusionmodules/lora.py	242
sat/sgm/modules/diffusionmodules/loss.py	98
sat/sgm/modules/diffusionmodules/model.py	544
sat/sgm/modules/diffusionmodules/openaimodel.py	979
sat/sgm/modules/diffusionmodules/sampling.py	608
sat/sgm/modules/diffusionmodules/sampling_utils.py	117
sat/sgm/modules/diffusionmodules/sigma_sampling.py	65
sat/sgm/modules/diffusionmodules/util.py	212
sat/sgm/modules/diffusionmodules/wrappers.py	33
sat/sgm/modules/distributions/__init__.py	1
sat/sgm/modules/distributions/distributions.py	66
sat/sgm/modules/ema.py	51
sat/sgm/modules/encoders/__init__.py	1
sat/sgm/modules/encoders/modules.py	242
sat/sgm/modules/traj_module.py	232
sat/sgm/modules/video_attention.py	249
sat/sgm/util.py	255
sat/sgm/webds.py	286
sat/train_video.py	348
sat/utils/flow_utils.py	154
sat/utils/misc.py	281
sat/utils/traj_utils.py	29
sat/vae_modules/attention.py	440
sat/vae_modules/autoencoder.py	534
sat/vae_modules/cp_enc_dec.py	712
sat/vae_modules/ema.py	51
sat/vae_modules/regularizers.py	83
sat/vae_modules/utils.py	267