Update README.md

README.md

@@ -49,6 +49,7 @@ This is a **DFloat11 losslessly compressed** version of the original `Qwen/Qwen-
     def parse_args():
         parser = argparse.ArgumentParser(description='Generate images using Qwen-Image model')
         parser.add_argument('--cpu_offload', action='store_true', help='Enable CPU offloading')
+        parser.add_argument('--no_pin_memory', action='store_true', help='Disable memory pinning')
         parser.add_argument('--prompt', type=str, default='A coffee shop entrance features a chalkboard sign reading "Qwen Coffee 😊 $2 per cup," with a neon light beside it displaying "通义千问". Next to it hangs a poster showing a beautiful Chinese woman, and beneath the poster is written "π≈3.1415926-53589793-23846264-33832795-02384197".',
                             help='Text prompt for image generation')
         parser.add_argument('--negative_prompt', type=str, default=' ',
@@ -82,6 +83,7 @@ This is a **DFloat11 losslessly compressed** version of the original `Qwen/Qwen-
         "DFloat11/Qwen-Image-DF11",
         device="cpu",
         cpu_offload=args.cpu_offload,
+        pin_memory=not args.no_pin_memory,
         bfloat16_model=transformer,
     )
 
@@ -134,6 +136,11 @@ This is a **DFloat11 losslessly compressed** version of the original `Qwen/Qwen-
     python qwen_image.py --cpu_offload
     ```
 
+    If you are getting out-of-memory errors, try disabling memory-pinning:
+    ```bash
+    python qwen_image.py --cpu_offload --no_pin_memory
+    ```
+
 
 ### 🔍 How It Works