github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-09-13 05:01:44 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix ckpt

Browse Source
This commit is contained in:
Fazzie
2023-02-03 15:34:54 +08:00
parent 6e0faa70e0
commit cad1f50512
7 changed files with 831 additions and 658 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
examples/images/diffusion/train_colossalai.sh
View File

@@ -2,4 +2,4 @@ HF_DATASETS_OFFLINE=1
TRANSFORMERS_OFFLINE=1
DIFFUSERS_OFFLINE=1
python main.py --logdir /tmp -t -b configs/train_colossalai.yaml
python main.py --logdir /tmp --train --base configs/Teyvat/train_colossalai_teyvat.yaml --ckpt diffuser_root_dir/512-base-ema.ckpt