|
|
|
# Download the pretrained checkpoints: |
|
|
|
First, make sure you have installed the huggingface CLI and modelscope CLI. |
|
|
|
```bash |
|
pip install -U "huggingface_hub[cli]" |
|
pip install modelscope |
|
``` |
|
|
|
|
|
### Download the pretrained DiT and VAE checkpoints: |
|
```bash |
|
hf download tencent/HunyuanImage-2.1 --local-dir ./ckpts |
|
``` |
|
|
|
### Downloading TextEncoders |
|
|
|
HunyuanImage uses an MLLM and a byT5 as text encoders. |
|
|
|
* **MLLM** |
|
|
|
HunyuanImage can be integrated with different MLLMs (including HunyuanMLLM and other open-source MLLM models). |
|
|
|
At this stage, we have not yet released the latest HunyuanMLLM. We recommend the users in community to use an open-source alternative, such as Qwen2.5-VL-7B-Instruct provided by Qwen Team, which can be downloaded by the following command: |
|
```bash |
|
hf download Qwen/Qwen2.5-VL-7B-Instruct --local-dir ./ckpts/text_encoder/llm |
|
``` |
|
|
|
* **ByT5 encoder** |
|
|
|
We use [Glyph-SDXL-v2](https://modelscope.cn/models/AI-ModelScope/Glyph-SDXL-v2) as our [byT5](https://github.com/google-research/byt5) encoder, which can be downloaded by the following command: |
|
|
|
```bash |
|
hf download google/byt5-small --local-dir ./ckpts/text_encoder/byt5-small |
|
modelscope download --model AI-ModelScope/Glyph-SDXL-v2 --local_dir ./ckpts/text_encoder/Glyph-SDXL-v2 |
|
``` |
|
You can also manually download the checkpoints from [here](https://modelscope.cn/models/AI-ModelScope/Glyph-SDXL-v2/files) and place them in the text_encoder folder like: |
|
``` |
|
ckpts |
|
βββ text_encoder |
|
βΒ Β βββ Glyph-SDXL-v2 |
|
βΒ Β βΒ Β βββ assets |
|
βΒ Β βΒ Β βΒ Β βββ color_idx.json |
|
βΒ Β βΒ Β βΒ Β βββ multilingual_10-lang_idx.json |
|
βΒ Β βΒ Β βΒ Β βββ ... |
|
βΒ Β βΒ Β βββ checkpoints |
|
βΒ Β βΒ Β Β Β βββ byt5_model.pt |
|
βΒ Β βΒ Β βββ ... |
|
β ββ ... |
|
ββ ... |
|
``` |
|
|
|
<details> |
|
|
|
<summary>π‘Tips for using hf/huggingface-cli (network problem)</summary> |
|
|
|
##### 1. Using HF-Mirror |
|
|
|
If you encounter slow download speeds in China, you can try a mirror to speed up the download process: |
|
|
|
```shell |
|
HF_ENDPOINT=https://hf-mirror.com hf download tencent/HunyuanImage-2.1 --local-dir ./ckpts |
|
``` |
|
|
|
##### 2. Resume Download |
|
|
|
`huggingface-cli` supports resuming downloads. If the download is interrupted, you can just rerun the download |
|
command to resume the download process. |
|
|
|
Note: If an `No such file or directory: 'ckpts/.huggingface/.gitignore.lock'` like error occurs during the download |
|
process, you can ignore the error and rerun the download command. |
|
|
|
</details> |
|
|