Upload janus_editing.sh with huggingface_hub

janus_editing.sh

@@ -13,13 +13,13 @@
 # limitations under the License.
 # ==============================================================================
 # Initialize variables
-export CUDA_VISIBLE_DEVICES=0,1,2,3
+export CUDA_VISIBLE_DEVICES=0,1,2
 MODEL_NAME_OR_PATH="/data5/czh/pretrain_models/Janus-Pro-1B"
 TRAIN_DATASETS="/home/czh/bxh/align-anything/projects/janus/example/supervised/text_to_image"
 TRAIN_DATA_FILE="Reasoning_Spatial_2.pt"
 #EVAL_DATASETS="/home/czh/bxh/align-anything/projects/janus/example/supervised/text_to_image"
 #EVAL_DATA_FILE="train_unify.pt"
-OUTPUT_DIR="/data5/czh/bxh/test_1"
+OUTPUT_DIR="/data5/czh/bxh/test_2"
 JANUS_REPO_PATH="/home/czh/bxh/Align_Anything_Janus" # change to your own path to Align_Anything_Janus
 
 export PYTHONPATH=$PYTHONPATH:$JANUS_REPO_PATH
@@ -36,7 +36,7 @@ deepspeed \
     --train_datasets ${TRAIN_DATASETS} \
     --train_data_files ${TRAIN_DATA_FILE} \
     --train_split train \
-    --learning_rate 1e-5 \
+    --learning_rate 5e-6 \
     --epochs 3 \
     --lr_scheduler_type cosine \
     --output_dir ${OUTPUT_DIR}