"...resnet50_tensorflow.git" did not exist on "4b3214c0a1c7e4526029bc14557ad26820d03c71"
run.sh 1.07 KB
Newer Older
wangwei990215's avatar
wangwei990215 committed
1
# source /public/home/chenxi/hopt/dtk24042-miopen.sh 
wangwei990215's avatar
wangwei990215 committed
2
export MODEL_NAME="/path/to/stable-diffusion-2-1-base"
wangwei990215's avatar
wangwei990215 committed
3
export OUTPUT_DIR="./output/sd2.1"
wangwei990215's avatar
wangwei990215 committed
4
export DATASET_NAME="path/to/train-00000-of-00001-566cc9b19d7203f8.parquet"
wangwei990215's avatar
wangwei990215 committed
5
#export DATASET_NAME="./pokemon-blip-captions"
wangwei990215's avatar
wangwei990215 committed
6
export HIP_VISIBLE_DEVICES=7
wangwei990215's avatar
wangwei990215 committed
7
8
9
#export LD_LIBRARY_PATH=/opt/rocblas-install/lib/:$LD_LIBRARY_PATH
#export PYTORCH_HIP_ALLOC_CONF=max_split_size_mb:25314

wangwei990215's avatar
wangwei990215 committed
10
python  train_text_to_image_lora.py \
wangwei990215's avatar
wangwei990215 committed
11
12
13
14
15
16
  --pretrained_model_name_or_path=$MODEL_NAME \
  --dataset_name=$DATASET_NAME \
  --dataloader_num_workers=8 \
  --resolution=960 \
  --center_crop \
  --random_flip \
wangwei990215's avatar
wangwei990215 committed
17
  --train_batch_size=8 \
wangwei990215's avatar
wangwei990215 committed
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
  --gradient_accumulation_steps=4 \
  --max_train_steps=101 \
  --learning_rate=1e-04 \
  --max_grad_norm=1 \
  --lr_scheduler="cosine" \
  --lr_warmup_steps=0 \
  --output_dir=${OUTPUT_DIR} \
  --report_to=wandb \
  --checkpointing_steps=50000 \
  --validation_prompt="A pokemon with blue eyes." \
  --report_to="wandb" \
  --seed=1337 \
  --mixed_precision="fp16" \
  --image_column="image" \
  --caption_column="text" \
  --max_grad_norm=1