Update README.md
Browse files
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=
|
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')
|