| export PATH=$PATH:/root/.local/bin | |
| export FLUX_DEV="/root/filesystem/Destyle_OmniStyle/weights/AI-ModelScope/FLUX.1-dev/flux1-dev.safetensors" | |
| export AE="/root/filesystem/Destyle_OmniStyle/weights/AI-ModelScope/FLUX.1-dev/ae.safetensors" | |
| export PYTORCH_CUDA_ALLOC_CONF=expandable_segments:True | |
| CUDA_VISIBLE_DEVICES=0,1,2,3,4,5,6,7 accelerate launch --multi_gpu --num_processes 8 train.py | |
| # CUDA_VISIBLE_DEVICES=0,1 accelerate launch --multi_gpu --num_processes 2 train.py |