townwish4git c8f09989da | 4 months ago | |
---|---|---|
.. | ||
configs | 4 months ago | |
gm | 4 months ago | |
README.md | 4 months ago | |
README_CN.md | 4 months ago | |
pangu_sampling.py | 4 months ago | |
prompts.txt | 4 months ago | |
requirement.txt | 4 months ago | |
run_sampling.sh | 4 months ago |
This folder contains PanGu Draw 3.0 models implemented with MindSpore.
In contrast to version 2.0, Pangu Draw 3.0 has been subject to experimentation and updates across various aspects, including multi-language support, diverse resolutions, improved image quality, and model scaling. This includes:
Dec 12, 2023
Support inference of PanGu Draw 3.0 model for text-to-image generation.
Please make sure the following frameworks are installed.
Install the dependent packages by running:
pip install -r requirements.txt
The text-to-image task of the Pangu model requires pre-training parameters for both the low timestamp model and the high timestamp model (The pre-training model parameters is coming soon).
Version | MindSpore Checkpoint |
---|---|
Pangu3-low-timestamp-model | pangu_low_timestamp-127da122.ckpt |
Pangu3-high-timestamp-model | pangu_high_timestamp-c6344411.ckpt |
After obtaining the weights, start the demo using:
# run txt2img on Ascend
export MS_PYNATIVE_GE=1
python demo/pangu/pangu_sampling.py \
--device_target "Ascend" \
--ms_amp_level "O2" \
--config "configs/inference/pangu_sd_xl_base.yaml" \
--high_solution \
--weight "path/to/low_timestamp_model.ckpt" \
--high_timestamp_weight "path/to/high_timestamp_model.ckpt" \
--prompt "Astronaut in a jungle, cold color palette, muted colors, detailed, 8k"
Note: sampled 40 steps by PanGu Draw 3.0 on Ascend 910*.
Fig1: "一幅中国水墨画:一叶轻舟漂泊在波光粼粼的湖面上,舟上的人正在饮酒放歌"
Fig2: "坐在海边看海浪的少年,黄昏"
Fig3: "Astronaut in a jungle, cold color palette, muted colors, detailed, 8k"
one for all, Optimal generator with No Exception
https://github.com/mindspore-lab/mindone
Python Text Markdown Shell
Dear OpenI User
Thank you for your continuous support to the Openl Qizhi Community AI Collaboration Platform. In order to protect your usage rights and ensure network security, we updated the Openl Qizhi Community AI Collaboration Platform Usage Agreement in January 2024. The updated agreement specifies that users are prohibited from using intranet penetration tools. After you click "Agree and continue", you can continue to use our services. Thank you for your cooperation and understanding.
For more agreement content, please refer to the《Openl Qizhi Community AI Collaboration Platform Usage Agreement》