Update train.sh

8d0fb2a5 · mashun1 · fa8e215a · 8d0fb2a5
Commit 8d0fb2a5 authored Nov 28, 2024 by mashun1
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

run/scripts/train.sh run/scripts/train.sh +1 -1

No files found.
--- a/run/scripts/train.sh
+++ b/run/scripts/train.sh
@@ -6,7 +6,7 @@ train_data_record_path="/home/vtryon/catvton_v0/datasets/train_data.jsonl"
 eval_data_record_path="/home/vtryon/catvton_v0/datasets/eval_data.jsonl"
 height=512
 width=384
-vae_subfolder="sd-vae-ft-mse"
+vae_subfolder="vae"
 weight_dtype="bf16"
 max_steps=32000
 logging_steps=1000