swap

Author: Saquib764

train/config/character_spatial_swap.yaml train/config/character_swap.yamltrain/config/character_spatial_swap.yaml renamed to train/config/character_swap.yaml

@@ -0,0 +1,14 @@
+# Specify the config file path and the GPU devices to use
+# export CUDA_VISIBLE_DEVICES=0,1
+
+# Specify the config file path
+export XFL_CONFIG=./train/config/character_swap.yaml
+
+export HF_HUB_CACHE=./cache
+# Specify the WANDB API key
+# export WANDB_API_KEY='YOUR_WANDB_API_KEY'
+
+echo $XFL_CONFIG
+export TOKENIZERS_PARALLELISM=true
+
+accelerate launch --main_process_port 41353 -m src.train.train