From ef8537d7928e9b5a73cca7c4a370e123172dc5f5 Mon Sep 17 00:00:00 2001
From: WangCunzheng <113966965+WangCunzheng@users.noreply.github.com>
Date: Mon, 18 Nov 2024 21:35:48 +0800
Subject: [PATCH] Create config.yaml

---
 train/SD15/config.yaml | 16 ++++++++++++++++
 1 file changed, 16 insertions(+)
 create mode 100644 train/SD15/config.yaml

diff --git a/train/SD15/config.yaml b/train/SD15/config.yaml
new file mode 100644
index 0000000..5cc2b3f
--- /dev/null
+++ 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