Update README.md

README.md

@@ -31,13 +31,18 @@ pip install -U hqq
 After installing the required library, you can run the following script: 
 
 ```python
+import torch
 from diffusers import FluxPipeline
 
 pipe = FluxPipeline.from_pretrained(
-    "HighCwu/FLUX.1-dev-bnb-hqq-4bit",
+    "HighCWu/FLUX.1-dev-bnb-hqq-4bit",
     torch_dtype=torch.bfloat16
 )
 
+# Use model cpu offload or all on cuda
+pipe.enable_model_cpu_offload()
+# pipe.to("cuda")
+
 prompt = "Baroque style, a lavish palace interior with ornate gilded ceilings, intricate tapestries, and dramatic lighting over a grand staircase."
 
 pipe_kwargs = {