Update README.md

README.md

@@ -33,31 +33,13 @@ trained using this [script](https://github.com/LambdaLabsML/examples/tree/main/s
 ```
 
 ```python
-import torch
-from diffusers import StableDiffusionPipeline
-from torch import autocast
-
-pipe = StableDiffusionPipeline.from_pretrained("wx44wx/sd-three-kingdoms-diffusers", torch_dtype=torch.float16)  
-pipe = pipe.to("cuda")
-
-prompt = "a man in armor"
-scale = 3
-n_samples = 4
-
-# Sometimes the nsfw checker is confused by the Pokémon images, you can disable
-# it at your own risk here
-disable_safety = False
-
-if disable_safety:
-  def null_safety(images, **kwargs):
-      return images, False
-  pipe.safety_checker = null_safety
-
-with autocast("cuda"):
-  images = pipe(n_samples*[prompt], guidance_scale=scale).images
-
-for idx, im in enumerate(images):
-  im.save(f"{idx:06}.png")
+ python scripts/txt2img.py \
+    --prompt 'a man in armor' \
+    --outdir 'outputs/generated_three_kingdoms_officers' \
+    --H 512 --W 512 \
+    --n_samples 4 \
+    --config 'configs/stable-diffusion/three_kingdoms.yaml' \
+    --ckpt ema-only-epoch=000144.ckpt
 ```
 
 ## Model description