Update README.md

README.md

@@ -23,4 +23,23 @@ save_to = "sayakpaul/flux.1-schell-int8wo-improved"
 transformer.push_to_hub(save_to, safe_serialization=False)
 ```
 
-Install `diffusers`, `huggingface_hub`, `ao` from the source.
+Install `diffusers`, `huggingface_hub`, `ao` from the source.
+
+Inference:
+
+```python
+import torch
+from diffusers import FluxTransformer2DModel, DiffusionPipeline
+
+dtype, device = torch.bfloat16, "cuda"
+ckpt_id = "black-forest-labs/FLUX.1-schnell"
+
+model = FluxTransformer2DModel.from_pretrained(
+    "sayakpaul/flux.1-schell-int8wo-improved", torch_dtype=dtype, use_safetensors=False
+)
+pipeline = DiffusionPipeline.from_pretrained(ckpt_id, transformer=model, torch_dtype=dtype).to("cuda")
+image = pipeline(
+	"cat", guidance_scale=0.0, num_inference_steps=4, max_sequence_length=256
+).images[0]
+image.save("flux_schnell_int8.png")
+```