Update README.md

README.md

@@ -34,7 +34,7 @@ import torch
 from diffusers import AmusedPipeline
 
 pipe = AmusedPipeline.from_pretrained(
-    "huggingface/amused-256", variant="fp16", torch_dtype=torch.float16
+    "amused/amused-256", variant="fp16", torch_dtype=torch.float16
 )
 pipe = pipe.to("cuda")
 
@@ -52,7 +52,7 @@ import torch
 from diffusers import AmusedPipeline
 
 pipe = AmusedPipeline.from_pretrained(
-    "huggingface/amused-512", variant="fp16", torch_dtype=torch.float16
+    "amused/amused-512", variant="fp16", torch_dtype=torch.float16
 )
 pipe = pipe.to("cuda")
 
@@ -73,7 +73,7 @@ from diffusers import AmusedImg2ImgPipeline
 from diffusers.utils import load_image
 
 pipe = AmusedImg2ImgPipeline.from_pretrained(
-    "huggingface/amused-256", variant="fp16", torch_dtype=torch.float16
+    "amused/amused-256", variant="fp16", torch_dtype=torch.float16
 )
 pipe = pipe.to("cuda")
 
@@ -100,7 +100,7 @@ from diffusers import AmusedImg2ImgPipeline
 from diffusers.utils import load_image
 
 pipe = AmusedImg2ImgPipeline.from_pretrained(
-    "huggingface/amused-512", variant="fp16", torch_dtype=torch.float16
+    "amused/amused-512", variant="fp16", torch_dtype=torch.float16
 )
 pipe = pipe.to("cuda")
 
@@ -130,7 +130,7 @@ from diffusers.utils import load_image
 from PIL import Image
 
 pipe = AmusedInpaintPipeline.from_pretrained(
-    "huggingface/amused-256", variant="fp16", torch_dtype=torch.float16
+    "amused/amused-256", variant="fp16", torch_dtype=torch.float16
 )
 pipe = pipe.to("cuda")
 
@@ -166,7 +166,7 @@ from diffusers import AmusedInpaintPipeline
 from diffusers.utils import load_image
 
 pipe = AmusedInpaintPipeline.from_pretrained(
-    "huggingface/amused-512", variant="fp16", torch_dtype=torch.float16
+    "amused/amused-512", variant="fp16", torch_dtype=torch.float16
 )
 pipe = pipe.to("cuda")
 
@@ -233,7 +233,7 @@ import torch
 from diffusers import AmusedPipeline
 
 pipe = AmusedPipeline.from_pretrained(
-    "huggingface/amused-256", variant="fp16", torch_dtype=torch.float16
+    "amused/amused-256", variant="fp16", torch_dtype=torch.float16
 )
 
 # HERE use torch.compile
@@ -317,7 +317,7 @@ accelerate launch training/training.py \
     --gradient_accumulation_steps <gradient accumulation steps> \
     --learning_rate 2e-5 \
     --use_8bit_adam \
-    --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 \
@@ -402,7 +402,7 @@ accelerate launch training/training.py \
     --train_batch_size <batch size> \
     --gradient_accumulation_steps <gradient accumulation steps> \
     --learning_rate 8e-5 \
-    --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 \
@@ -474,7 +474,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-512 \
+    --pretrained_model_name_or_path amused/amused-512 \
     --instance_data_dataset  'monadical-labs/minecraft-preview' \
     --prompt_prefix 'minecraft ' \
     --image_key image \
@@ -515,7 +515,7 @@ accelerate launch ./training/training.py \
     --mixed_precision fp16 \
     --report_to wandb \
     --use_lora \
-    --pretrained_model_name_or_path huggingface/amused-256 \
+    --pretrained_model_name_or_path amused/amused-256 \
     --train_batch_size 1 \
     --lr_scheduler constant \
     --learning_rate 4e-4 \
@@ -545,7 +545,7 @@ accelerate launch ./training/training.py \
     --mixed_precision fp16 \
     --report_to wandb \
     --use_lora \
-    --pretrained_model_name_or_path huggingface/amused-512 \
+    --pretrained_model_name_or_path amused/amused-512 \
     --train_batch_size 1 \
     --lr_scheduler constant \
     --learning_rate 1e-3 \