train/script/train_spatial_alignment_rotation.sh

# *[Specify the config file path and the GPU devices to use]
# export CUDA_VISIBLE_DEVICES=0,1

# *[Specify the config file path]
export OMINI_CONFIG=./train/config/spatial_alignment_rotation.yaml

# *[Specify the WANDB API key]
export WANDB_API_KEY='675d7815d8ad1cf89954fd5415071f18007386fd'

echo $OMINI_CONFIG
export TOKENIZERS_PARALLELISM=true

accelerate launch --main_process_port 41353 -m omini.train_flux.train_spatial_alignment_rotation