Update README.md

README.md

@@ -80,7 +80,7 @@ pipe = pipe.to("cuda")
 prompt = "apple watercolor"
 input_image = (
     load_image(
-        "https://raw.githubusercontent.com/huggingface/amused/main/assets/image2image_256_orig.png"
+        "https://raw.githubusercontent.com/amused/amused/main/assets/image2image_256_orig.png"
     )
     .resize((256, 256))
     .convert("RGB")
@@ -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 \