change run.sh

src/run.sh

@@ -28,14 +28,14 @@ export LOGGING_STEPS=500
 export EVAL_STEPS=2500
 export SAVE_STEPS=2500
 
-python src/run_clm.py \
+python src/run_clm_flax.py \
     --output_dir="$OUTPUT_DIR"  \
     --model_type="$MODEL_TYPE" \
     --config_name="$CONFIG_NAME" \
     --tokenizer_name="$TOKENIZER_NAME" \
     --dataset_name="$DATASET_NAME" \
     --dataset_config_name="$DATASET_CONFIG_NAME" \
-    --max_seq_length="$MAX_SEQUENCE_LENGTH" \
+    --block_size=$MAX_SEQUENCE_LENGTH \
     --per_device_train_batch_size=$PER_DEVICE_TRAIN_BATCH_SIZE \
     --per_device_eval_batch_size=$PER_DEVICE_EVAL_BATCH_SIZE \
     --num_train_epochs=$NUM_TRAIN_EPOCHS \
@@ -47,4 +47,4 @@ python src/run_clm.py \
     --do_train \
     --do_eval \
     --overwrite_output_dir \
-    --push_to_hub
+    --push_to_hub