github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-09-18 07:31:19 +00:00
Code Issues Packages Projects Releases Wiki Activity

[example] Change some training settings for diffusion (#2195)

Browse Source
This commit is contained in:
BlueRum
2022-12-26 15:22:20 +08:00
committed by GitHub
parent 2458659919
commit 6642cebdbe
6 changed files with 30 additions and 24 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
examples/images/diffusion/train_colossalai.sh Executable file
View File

@@ -0,0 +1,5 @@
HF_DATASETS_OFFLINE=1
TRANSFORMERS_OFFLINE=1
DIFFUSERS_OFFLINE=1
python main.py --logdir /tmp -t -b /configs/train_colossalai.yaml