Create config.yaml

RedAIGC · Nov 18, 2024 · ef8537d · ef8537d
1 parent d081e87
commit ef8537d
Showing 1 changed file with 16 additions and 0 deletions.
diff --git a/train/SD15/config.yaml b/train/SD15/config.yaml
@@ -0,0 +1,16 @@
+compute_environment: LOCAL_MACHINE
+debug: false
+distributed_type: MULTI_GPU
+downcast_bf16: 'no'
+gpu_ids: all
+machine_rank: 0
+main_training_function: main
+mixed_precision: fp16
+num_machines: 1
+num_processes: 8
+rdzv_backend: static
+same_network: true
+tpu_env: []
+tpu_use_cluster: false
+tpu_use_sudo: false
+use_cpu: false