Update README.md

README.md

@@ -59,18 +59,17 @@ Please follow the instructions in [mit-han-lab/nunchaku](https://github.com/mit-
 
 ```python
 import torch
+from diffusers import FluxPipeline
 
-from nunchaku.pipelines import flux as nunchaku_flux
+from nunchaku.models.transformer_flux import NunchakuFluxTransformer2dModel
 
-pipeline = nunchaku_flux.from_pretrained(
-    "black-forest-labs/FLUX.1-dev",
-    torch_dtype=torch.bfloat16,
-    qmodel_path="mit-han-lab/svdq-int4-flux.1-dev",  # download from Huggingface
+transformer = NunchakuFluxTransformer2dModel.from_pretrained("mit-han-lab/svdq-int4-flux.1-dev")
+pipeline = FluxPipeline.from_pretrained(
+    "black-forest-labs/FLUX.1-dev", transformer=transformer, torch_dtype=torch.bfloat16
 ).to("cuda")
-pipeline.transformer.nunchaku_update_params(mit-han-lab/svdquant-models/svdq-flux.1-dev-lora-anime.safetensors)
-pipeline.transformer.nunchaku_set_lora_scale(1)
-image = pipeline("a dog wearing a wizard hat", num_inference_steps=28, guidance_scale=3.5).images[0]
+image = pipeline("A cat holding a sign that says hello world", num_inference_steps=50, guidance_scale=3.5).images[0]
 image.save("example.png")
+
 ```
 
 ### Comfy UI