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

[exmaple] fix dreamblooth format (#2315)

Browse Source
This commit is contained in:
Fazzie-Maqianli
2023-01-04 16:20:00 +08:00
committed by GitHub
parent da1c47f060
commit a9b27b9265
5 changed files with 90 additions and 101 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
examples/images/diffusion/configs/Teyvat/train_colossalai_teyvat.yaml
View File

@@ -108,7 +108,7 @@ lightning:
params:
use_chunk: True
enable_distributed_storage: True
placement_policy: auto
placement_policy: cuda
force_outputs_fp32: true
log_every_n_steps: 2

2
examples/images/diffusion/configs/train_colossalai.yaml
View File

@@ -105,7 +105,7 @@ lightning:
params:
use_chunk: True
enable_distributed_storage: True
placement_policy: auto
placement_policy: cuda
force_outputs_fp32: true
log_every_n_steps: 2

2
examples/images/diffusion/configs/train_colossalai_cifar10.yaml
View File

@@ -109,7 +109,7 @@ lightning:
params:
use_chunk: True
enable_distributed_storage: True
placement_policy: auto
placement_policy: cuda
force_outputs_fp32: true
log_every_n_steps: 2

2
examples/images/diffusion/configs/train_pokemon.yaml
View File

@@ -102,7 +102,7 @@ lightning:
params:
use_chunk: True
enable_distributed_storage: True
placement_policy: auto
placement_policy: cuda
force_outputs_fp32: true
log_every_n_steps: 2