├── .gitignore ├── README.md ├── config_OneVAE ├── OneVAE_64000_FSQ_161616_33_mt2_fe.yaml ├── OneVAE_64000_FSQ_81616_33_mt2_fe.yaml └── OneVAE_64000_FSQ_888_33_mt2.yaml ├── eval_tokenizer.py ├── models ├── __init__.py ├── image_base_model.py ├── model_utils │ ├── ema.py │ ├── save.py │ └── utils.py └── video_base_model.py ├── modules ├── onevae │ ├── vae161616_hybrid.py │ ├── vae81616_hybrid.py │ └── vae888.py ├── posemb_layers.py ├── positional_embeddings.py ├── util.py └── vqvae │ ├── enc_dec_llamagen.py │ ├── quantize.py │ ├── quantize_FSQ.py │ ├── quantize_IBQ.py │ └── quantize_simvq.py └── util.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/OneVAE/HEAD/.gitignore -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/OneVAE/HEAD/README.md -------------------------------------------------------------------------------- /config_OneVAE/OneVAE_64000_FSQ_161616_33_mt2_fe.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/OneVAE/HEAD/config_OneVAE/OneVAE_64000_FSQ_161616_33_mt2_fe.yaml -------------------------------------------------------------------------------- /config_OneVAE/OneVAE_64000_FSQ_81616_33_mt2_fe.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/OneVAE/HEAD/config_OneVAE/OneVAE_64000_FSQ_81616_33_mt2_fe.yaml -------------------------------------------------------------------------------- /config_OneVAE/OneVAE_64000_FSQ_888_33_mt2.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/OneVAE/HEAD/config_OneVAE/OneVAE_64000_FSQ_888_33_mt2.yaml -------------------------------------------------------------------------------- /eval_tokenizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/OneVAE/HEAD/eval_tokenizer.py -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /models/image_base_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/OneVAE/HEAD/models/image_base_model.py -------------------------------------------------------------------------------- /models/model_utils/ema.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/OneVAE/HEAD/models/model_utils/ema.py -------------------------------------------------------------------------------- /models/model_utils/save.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/OneVAE/HEAD/models/model_utils/save.py -------------------------------------------------------------------------------- /models/model_utils/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/OneVAE/HEAD/models/model_utils/utils.py -------------------------------------------------------------------------------- /models/video_base_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/OneVAE/HEAD/models/video_base_model.py -------------------------------------------------------------------------------- /modules/onevae/vae161616_hybrid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/OneVAE/HEAD/modules/onevae/vae161616_hybrid.py -------------------------------------------------------------------------------- /modules/onevae/vae81616_hybrid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/OneVAE/HEAD/modules/onevae/vae81616_hybrid.py -------------------------------------------------------------------------------- /modules/onevae/vae888.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/OneVAE/HEAD/modules/onevae/vae888.py -------------------------------------------------------------------------------- /modules/posemb_layers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/OneVAE/HEAD/modules/posemb_layers.py -------------------------------------------------------------------------------- /modules/positional_embeddings.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/OneVAE/HEAD/modules/positional_embeddings.py -------------------------------------------------------------------------------- /modules/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/OneVAE/HEAD/modules/util.py -------------------------------------------------------------------------------- /modules/vqvae/enc_dec_llamagen.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/OneVAE/HEAD/modules/vqvae/enc_dec_llamagen.py -------------------------------------------------------------------------------- /modules/vqvae/quantize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/OneVAE/HEAD/modules/vqvae/quantize.py -------------------------------------------------------------------------------- /modules/vqvae/quantize_FSQ.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/OneVAE/HEAD/modules/vqvae/quantize_FSQ.py -------------------------------------------------------------------------------- /modules/vqvae/quantize_IBQ.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/OneVAE/HEAD/modules/vqvae/quantize_IBQ.py -------------------------------------------------------------------------------- /modules/vqvae/quantize_simvq.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/OneVAE/HEAD/modules/vqvae/quantize_simvq.py -------------------------------------------------------------------------------- /util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/OneVAE/HEAD/util.py --------------------------------------------------------------------------------