LeanQuant commited on
Commit
4f8f857
·
verified ·
1 Parent(s): 729204d

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -51,7 +51,7 @@ This is a **DFloat11 losslessly compressed** version of the original `Qwen/Qwen-
51
  def parse_args():
52
  parser = argparse.ArgumentParser(description='Edit images using Qwen-Image-Edit model')
53
  parser.add_argument('--cpu_offload', action='store_true', help='Enable CPU offloading')
54
- parser.add_argument('--cpu_offload_blocks', type=int, default=16, help='Number of transformer blocks to offload to CPU')
55
  parser.add_argument('--no_pin_memory', action='store_true', help='Disable memory pinning')
56
  parser.add_argument('--image', type=str, default="https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/diffusers/cat.png",
57
  help='Path to input image or URL')
 
51
  def parse_args():
52
  parser = argparse.ArgumentParser(description='Edit images using Qwen-Image-Edit model')
53
  parser.add_argument('--cpu_offload', action='store_true', help='Enable CPU offloading')
54
+ parser.add_argument('--cpu_offload_blocks', type=int, default=30, help='Number of transformer blocks to offload to CPU')
55
  parser.add_argument('--no_pin_memory', action='store_true', help='Disable memory pinning')
56
  parser.add_argument('--image', type=str, default="https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/diffusers/cat.png",
57
  help='Path to input image or URL')