Skip to content

Commit

Permalink
remove: unnecessary arg in generate_image
Browse files Browse the repository at this point in the history
  • Loading branch information
EduardoPach committed Oct 28, 2024
1 parent 5f8a185 commit 1d05fce
Showing 1 changed file with 1 addition and 9 deletions.
10 changes: 1 addition & 9 deletions python/src/diffusionkit/mlx/scripts/generate_images.py
Original file line number Diff line number Diff line change
Expand Up @@ -111,11 +111,7 @@ def cli():
type=str,
help="Path to the local mmdit checkpoint.",
)
parser.add_argument(
"--quantized",
action="store_true",
help="Use 4-bit quantized model if available",
)

args = parser.parse_args()

args.w16 = True
Expand All @@ -133,10 +129,6 @@ def cli():
if args.denoise < 0.0 or args.denoise > 1.0:
raise ValueError("Denoising factor must be between 0.0 and 1.0")

if args.quantized and "4bit-quantized" not in args.model_version:
args.model_version += "-4bit-quantized"
logger.info(f"Using 4-bit quantized model: {args.model_version}")

shift = args.shift or SHIFT[args.model_version]
pipeline_class = FluxPipeline if "FLUX" in args.model_version else DiffusionPipeline

Expand Down

0 comments on commit 1d05fce

Please sign in to comment.