Update README.md

README.md

@@ -18,10 +18,21 @@ Make everything as Yarn Art Syle.
 
 ```python
 !pip install -U diffusers bitsandbytes
-from diffusers import AutoPipelineForText2Image
+
+from diffusers import AutoPipelineForText2Image, BitsAndBytesConfig
+import torch
 repo_id = "markod0925/yarn_flux"
 
-pipeline = AutoPipelineForText2Image.from_pretrained(repo_id)
+bnb_4bit_compute_dtype = torch.float16
+nf4_config = BitsAndBytesConfig(
+    load_in_4bit=True,
+    bnb_4bit_quant_type="nf4",
+    bnb_4bit_compute_dtype=bnb_4bit_compute_dtype,
+)
+
+pipeline = AutoPipelineForText2Image.from_pretrained(repo_id,
+                                                     torch_dtype=bnb_4bit_compute_dtype
+)
 pipeline.enable_model_cpu_offload()
 
 image = pipeline(