train/script/train_compact_token_representation.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/compact_token_representation.yaml

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

echo $OMINI_CONFIG
export TOKENIZERS_PARALLELISM=true

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