Update README.md

README.md

@@ -276,7 +276,7 @@ accelerate launch training/training.py \
     --train_batch_size <batch size> \
     --gradient_accumulation_steps <gradient accumulation steps> \
     --learning_rate 1e-4 \
-    --pretrained_model_name_or_path huggingface/amused-256 \
+    --pretrained_model_name_or_path amused/amused-256 \
     --instance_data_dataset  'm1guelpf/nouns' \
     --image_key image \
     --prompt_key text \
@@ -356,7 +356,7 @@ accelerate launch training/training.py \
     --gradient_accumulation_steps <gradient accumulation steps> \
     --learning_rate 8e-4 \
     --use_lora \
-    --pretrained_model_name_or_path huggingface/amused-256 \
+    --pretrained_model_name_or_path amused/amused-256 \
     --instance_data_dataset  'm1guelpf/nouns' \
     --image_key image \
     --prompt_key text \
@@ -438,7 +438,7 @@ accelerate launch training/training.py \
     --train_batch_size <batch size> \
     --gradient_accumulation_steps <gradient accumulation steps> \
     --learning_rate 5e-6 \
-    --pretrained_model_name_or_path huggingface/amused-512 \
+    --pretrained_model_name_or_path amused/amused-512 \
     --instance_data_dataset  'monadical-labs/minecraft-preview' \
     --prompt_prefix 'minecraft ' \
     --image_key image \