├── .gitignore ├── LICENSE ├── README.md ├── assets ├── cute_girl.jpg └── logo2.png ├── cluster_label.npy ├── cog.yaml ├── coglm_strategy.py ├── cogview2_completion.py ├── cogview2_text2image.py ├── comp_pipeline ├── __init__.py ├── base_completion.py └── patch_completion.py ├── input.txt ├── input_comp.txt ├── predict.py ├── pretrain_coglm.py ├── requirements.txt ├── sr_pipeline ├── __init__.py ├── direct_sr.py ├── dsr_model.py ├── dsr_sampling.py ├── iterative_sr.py ├── itersr_model.py ├── itersr_sampling.py └── sr_group.py ├── text2image.sh └── text_guided_completion.sh /.gitignore: -------------------------------------------------------------------------------- 1 | samples* 2 | __pycache__ 3 | -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogView2/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogView2/HEAD/README.md -------------------------------------------------------------------------------- /assets/cute_girl.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogView2/HEAD/assets/cute_girl.jpg -------------------------------------------------------------------------------- /assets/logo2.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogView2/HEAD/assets/logo2.png -------------------------------------------------------------------------------- /cluster_label.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogView2/HEAD/cluster_label.npy -------------------------------------------------------------------------------- /cog.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogView2/HEAD/cog.yaml -------------------------------------------------------------------------------- /coglm_strategy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogView2/HEAD/coglm_strategy.py -------------------------------------------------------------------------------- /cogview2_completion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogView2/HEAD/cogview2_completion.py -------------------------------------------------------------------------------- /cogview2_text2image.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogView2/HEAD/cogview2_text2image.py -------------------------------------------------------------------------------- /comp_pipeline/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogView2/HEAD/comp_pipeline/__init__.py -------------------------------------------------------------------------------- /comp_pipeline/base_completion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogView2/HEAD/comp_pipeline/base_completion.py -------------------------------------------------------------------------------- /comp_pipeline/patch_completion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogView2/HEAD/comp_pipeline/patch_completion.py -------------------------------------------------------------------------------- /input.txt: -------------------------------------------------------------------------------- 1 | 0 a tiger wearing VR glasses -------------------------------------------------------------------------------- /input_comp.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogView2/HEAD/input_comp.txt -------------------------------------------------------------------------------- /predict.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogView2/HEAD/predict.py -------------------------------------------------------------------------------- /pretrain_coglm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogView2/HEAD/pretrain_coglm.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- 1 | SwissArmyTransformer>=0.2 2 | icetk 3 | -------------------------------------------------------------------------------- /sr_pipeline/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogView2/HEAD/sr_pipeline/__init__.py -------------------------------------------------------------------------------- /sr_pipeline/direct_sr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogView2/HEAD/sr_pipeline/direct_sr.py -------------------------------------------------------------------------------- /sr_pipeline/dsr_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogView2/HEAD/sr_pipeline/dsr_model.py -------------------------------------------------------------------------------- /sr_pipeline/dsr_sampling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogView2/HEAD/sr_pipeline/dsr_sampling.py -------------------------------------------------------------------------------- /sr_pipeline/iterative_sr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogView2/HEAD/sr_pipeline/iterative_sr.py -------------------------------------------------------------------------------- /sr_pipeline/itersr_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogView2/HEAD/sr_pipeline/itersr_model.py -------------------------------------------------------------------------------- /sr_pipeline/itersr_sampling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogView2/HEAD/sr_pipeline/itersr_sampling.py -------------------------------------------------------------------------------- /sr_pipeline/sr_group.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogView2/HEAD/sr_pipeline/sr_group.py -------------------------------------------------------------------------------- /text2image.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogView2/HEAD/text2image.sh -------------------------------------------------------------------------------- /text_guided_completion.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zai-org/CogView2/HEAD/text_guided_completion.sh --------------------------------------------------------------------------------