├── .gitattributes ├── .gitignore ├── CITATION.cff ├── LICENSE ├── README.md ├── assets └── v1.0 │ ├── generated_offset100.gif │ ├── generated_offset1000100.gif │ ├── generated_offset100100.gif │ ├── generated_offset125100.gif │ ├── generated_offset150100.gif │ ├── generated_offset175100.gif │ ├── generated_offset225100.gif │ ├── generated_offset250100.gif │ ├── generated_offset25100.gif │ ├── generated_offset275100.gif │ ├── generated_offset300100.gif │ ├── generated_offset350100.gif │ ├── generated_offset375100.gif │ ├── generated_offset400100.gif │ ├── generated_offset450100.gif │ ├── generated_offset475100.gif │ ├── generated_offset500100.gif │ ├── generated_offset50100.gif │ ├── generated_offset525100.gif │ ├── generated_offset600100.gif │ ├── generated_offset625100.gif │ ├── generated_offset650100.gif │ ├── generated_offset675100.gif │ ├── generated_offset700100.gif │ ├── generated_offset725100.gif │ ├── generated_offset750100.gif │ ├── generated_offset775100.gif │ ├── generated_offset800100.gif │ ├── generated_offset825100.gif │ ├── generated_offset850100.gif │ ├── generated_offset875100.gif │ ├── generated_offset900100.gif │ ├── generated_offset925100.gif │ ├── generated_offset950100.gif │ └── generated_offset975100.gif ├── data.py ├── eval_utils.py ├── genie ├── __init__.py ├── attention.py ├── config.py ├── configs │ └── magvit_n32_h8_d256.json ├── evaluate.py ├── factorization_utils.py ├── generate.py ├── st_mask_git.py ├── st_transformer.py └── utils.py ├── magvit2 ├── LICENSE ├── README.md ├── __init__.py ├── config.py ├── models │ ├── __init__.py │ └── lfqgan.py ├── modules │ ├── __init__.py │ ├── autoencoder │ │ └── lpips │ │ │ └── vgg.pth │ ├── diffusionmodules │ │ ├── __init__.py │ │ └── improved_model.py │ ├── discriminator │ │ ├── __init__.py │ │ └── model.py │ ├── ema.py │ ├── losses │ │ ├── __init__.py │ │ ├── lpips.py │ │ └── vqperceptual.py │ ├── scheduler │ │ ├── __init__.py │ │ └── lr_scheduler.py │ ├── util.py │ └── vqvae │ │ ├── __init__.py │ │ ├── lookup_free_quantize.py │ │ └── quantize.py └── util.py ├── requirements.txt ├── setup.py ├── test_attention.py ├── train.py └── visualize.py /.gitattributes: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/1x-technologies/1xgpt/HEAD/.gitattributes -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/1x-technologies/1xgpt/HEAD/.gitignore -------------------------------------------------------------------------------- /CITATION.cff: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/1x-technologies/1xgpt/HEAD/CITATION.cff -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/1x-technologies/1xgpt/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/1x-technologies/1xgpt/HEAD/README.md -------------------------------------------------------------------------------- /assets/v1.0/generated_offset100.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/1x-technologies/1xgpt/HEAD/assets/v1.0/generated_offset100.gif -------------------------------------------------------------------------------- /assets/v1.0/generated_offset1000100.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/1x-technologies/1xgpt/HEAD/assets/v1.0/generated_offset1000100.gif -------------------------------------------------------------------------------- /assets/v1.0/generated_offset100100.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/1x-technologies/1xgpt/HEAD/assets/v1.0/generated_offset100100.gif -------------------------------------------------------------------------------- /assets/v1.0/generated_offset125100.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/1x-technologies/1xgpt/HEAD/assets/v1.0/generated_offset125100.gif -------------------------------------------------------------------------------- /assets/v1.0/generated_offset150100.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/1x-technologies/1xgpt/HEAD/assets/v1.0/generated_offset150100.gif -------------------------------------------------------------------------------- /assets/v1.0/generated_offset175100.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/1x-technologies/1xgpt/HEAD/assets/v1.0/generated_offset175100.gif -------------------------------------------------------------------------------- /assets/v1.0/generated_offset225100.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/1x-technologies/1xgpt/HEAD/assets/v1.0/generated_offset225100.gif -------------------------------------------------------------------------------- /assets/v1.0/generated_offset250100.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/1x-technologies/1xgpt/HEAD/assets/v1.0/generated_offset250100.gif -------------------------------------------------------------------------------- /assets/v1.0/generated_offset25100.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/1x-technologies/1xgpt/HEAD/assets/v1.0/generated_offset25100.gif -------------------------------------------------------------------------------- /assets/v1.0/generated_offset275100.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/1x-technologies/1xgpt/HEAD/assets/v1.0/generated_offset275100.gif -------------------------------------------------------------------------------- /assets/v1.0/generated_offset300100.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/1x-technologies/1xgpt/HEAD/assets/v1.0/generated_offset300100.gif -------------------------------------------------------------------------------- /assets/v1.0/generated_offset350100.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/1x-technologies/1xgpt/HEAD/assets/v1.0/generated_offset350100.gif -------------------------------------------------------------------------------- /assets/v1.0/generated_offset375100.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/1x-technologies/1xgpt/HEAD/assets/v1.0/generated_offset375100.gif -------------------------------------------------------------------------------- /assets/v1.0/generated_offset400100.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/1x-technologies/1xgpt/HEAD/assets/v1.0/generated_offset400100.gif -------------------------------------------------------------------------------- /assets/v1.0/generated_offset450100.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/1x-technologies/1xgpt/HEAD/assets/v1.0/generated_offset450100.gif -------------------------------------------------------------------------------- /assets/v1.0/generated_offset475100.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/1x-technologies/1xgpt/HEAD/assets/v1.0/generated_offset475100.gif -------------------------------------------------------------------------------- /assets/v1.0/generated_offset500100.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/1x-technologies/1xgpt/HEAD/assets/v1.0/generated_offset500100.gif -------------------------------------------------------------------------------- /assets/v1.0/generated_offset50100.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/1x-technologies/1xgpt/HEAD/assets/v1.0/generated_offset50100.gif -------------------------------------------------------------------------------- /assets/v1.0/generated_offset525100.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/1x-technologies/1xgpt/HEAD/assets/v1.0/generated_offset525100.gif -------------------------------------------------------------------------------- /assets/v1.0/generated_offset600100.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/1x-technologies/1xgpt/HEAD/assets/v1.0/generated_offset600100.gif -------------------------------------------------------------------------------- /assets/v1.0/generated_offset625100.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/1x-technologies/1xgpt/HEAD/assets/v1.0/generated_offset625100.gif -------------------------------------------------------------------------------- /assets/v1.0/generated_offset650100.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/1x-technologies/1xgpt/HEAD/assets/v1.0/generated_offset650100.gif -------------------------------------------------------------------------------- /assets/v1.0/generated_offset675100.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/1x-technologies/1xgpt/HEAD/assets/v1.0/generated_offset675100.gif -------------------------------------------------------------------------------- /assets/v1.0/generated_offset700100.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/1x-technologies/1xgpt/HEAD/assets/v1.0/generated_offset700100.gif -------------------------------------------------------------------------------- /assets/v1.0/generated_offset725100.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/1x-technologies/1xgpt/HEAD/assets/v1.0/generated_offset725100.gif -------------------------------------------------------------------------------- /assets/v1.0/generated_offset750100.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/1x-technologies/1xgpt/HEAD/assets/v1.0/generated_offset750100.gif -------------------------------------------------------------------------------- /assets/v1.0/generated_offset775100.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/1x-technologies/1xgpt/HEAD/assets/v1.0/generated_offset775100.gif -------------------------------------------------------------------------------- /assets/v1.0/generated_offset800100.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/1x-technologies/1xgpt/HEAD/assets/v1.0/generated_offset800100.gif -------------------------------------------------------------------------------- /assets/v1.0/generated_offset825100.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/1x-technologies/1xgpt/HEAD/assets/v1.0/generated_offset825100.gif -------------------------------------------------------------------------------- /assets/v1.0/generated_offset850100.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/1x-technologies/1xgpt/HEAD/assets/v1.0/generated_offset850100.gif -------------------------------------------------------------------------------- /assets/v1.0/generated_offset875100.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/1x-technologies/1xgpt/HEAD/assets/v1.0/generated_offset875100.gif -------------------------------------------------------------------------------- /assets/v1.0/generated_offset900100.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/1x-technologies/1xgpt/HEAD/assets/v1.0/generated_offset900100.gif -------------------------------------------------------------------------------- /assets/v1.0/generated_offset925100.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/1x-technologies/1xgpt/HEAD/assets/v1.0/generated_offset925100.gif -------------------------------------------------------------------------------- /assets/v1.0/generated_offset950100.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/1x-technologies/1xgpt/HEAD/assets/v1.0/generated_offset950100.gif -------------------------------------------------------------------------------- /assets/v1.0/generated_offset975100.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/1x-technologies/1xgpt/HEAD/assets/v1.0/generated_offset975100.gif -------------------------------------------------------------------------------- /data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/1x-technologies/1xgpt/HEAD/data.py -------------------------------------------------------------------------------- /eval_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/1x-technologies/1xgpt/HEAD/eval_utils.py -------------------------------------------------------------------------------- /genie/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /genie/attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/1x-technologies/1xgpt/HEAD/genie/attention.py -------------------------------------------------------------------------------- /genie/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/1x-technologies/1xgpt/HEAD/genie/config.py -------------------------------------------------------------------------------- /genie/configs/magvit_n32_h8_d256.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/1x-technologies/1xgpt/HEAD/genie/configs/magvit_n32_h8_d256.json -------------------------------------------------------------------------------- /genie/evaluate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/1x-technologies/1xgpt/HEAD/genie/evaluate.py -------------------------------------------------------------------------------- /genie/factorization_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/1x-technologies/1xgpt/HEAD/genie/factorization_utils.py -------------------------------------------------------------------------------- /genie/generate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/1x-technologies/1xgpt/HEAD/genie/generate.py -------------------------------------------------------------------------------- /genie/st_mask_git.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/1x-technologies/1xgpt/HEAD/genie/st_mask_git.py -------------------------------------------------------------------------------- /genie/st_transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/1x-technologies/1xgpt/HEAD/genie/st_transformer.py -------------------------------------------------------------------------------- /genie/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/1x-technologies/1xgpt/HEAD/genie/utils.py -------------------------------------------------------------------------------- /magvit2/LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/1x-technologies/1xgpt/HEAD/magvit2/LICENSE -------------------------------------------------------------------------------- /magvit2/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/1x-technologies/1xgpt/HEAD/magvit2/README.md -------------------------------------------------------------------------------- /magvit2/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /magvit2/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/1x-technologies/1xgpt/HEAD/magvit2/config.py -------------------------------------------------------------------------------- /magvit2/models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /magvit2/models/lfqgan.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/1x-technologies/1xgpt/HEAD/magvit2/models/lfqgan.py -------------------------------------------------------------------------------- /magvit2/modules/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /magvit2/modules/autoencoder/lpips/vgg.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/1x-technologies/1xgpt/HEAD/magvit2/modules/autoencoder/lpips/vgg.pth -------------------------------------------------------------------------------- /magvit2/modules/diffusionmodules/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /magvit2/modules/diffusionmodules/improved_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/1x-technologies/1xgpt/HEAD/magvit2/modules/diffusionmodules/improved_model.py -------------------------------------------------------------------------------- /magvit2/modules/discriminator/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /magvit2/modules/discriminator/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/1x-technologies/1xgpt/HEAD/magvit2/modules/discriminator/model.py -------------------------------------------------------------------------------- /magvit2/modules/ema.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/1x-technologies/1xgpt/HEAD/magvit2/modules/ema.py -------------------------------------------------------------------------------- /magvit2/modules/losses/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/1x-technologies/1xgpt/HEAD/magvit2/modules/losses/__init__.py -------------------------------------------------------------------------------- /magvit2/modules/losses/lpips.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/1x-technologies/1xgpt/HEAD/magvit2/modules/losses/lpips.py -------------------------------------------------------------------------------- /magvit2/modules/losses/vqperceptual.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/1x-technologies/1xgpt/HEAD/magvit2/modules/losses/vqperceptual.py -------------------------------------------------------------------------------- /magvit2/modules/scheduler/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /magvit2/modules/scheduler/lr_scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/1x-technologies/1xgpt/HEAD/magvit2/modules/scheduler/lr_scheduler.py -------------------------------------------------------------------------------- /magvit2/modules/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/1x-technologies/1xgpt/HEAD/magvit2/modules/util.py -------------------------------------------------------------------------------- /magvit2/modules/vqvae/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /magvit2/modules/vqvae/lookup_free_quantize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/1x-technologies/1xgpt/HEAD/magvit2/modules/vqvae/lookup_free_quantize.py -------------------------------------------------------------------------------- /magvit2/modules/vqvae/quantize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/1x-technologies/1xgpt/HEAD/magvit2/modules/vqvae/quantize.py -------------------------------------------------------------------------------- /magvit2/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/1x-technologies/1xgpt/HEAD/magvit2/util.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/1x-technologies/1xgpt/HEAD/requirements.txt -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/1x-technologies/1xgpt/HEAD/setup.py -------------------------------------------------------------------------------- /test_attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/1x-technologies/1xgpt/HEAD/test_attention.py -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/1x-technologies/1xgpt/HEAD/train.py -------------------------------------------------------------------------------- /visualize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/1x-technologies/1xgpt/HEAD/visualize.py --------------------------------------------------------------------------------