Skip to content

Commit 1d05fce

Browse files
committed
remove: unnecessary arg in generate_image
1 parent 5f8a185 commit 1d05fce

File tree

1 file changed

+1
-9
lines changed

1 file changed

+1
-9
lines changed

python/src/diffusionkit/mlx/scripts/generate_images.py

Lines changed: 1 addition & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -111,11 +111,7 @@ def cli():
111111
type=str,
112112
help="Path to the local mmdit checkpoint.",
113113
)
114-
parser.add_argument(
115-
"--quantized",
116-
action="store_true",
117-
help="Use 4-bit quantized model if available",
118-
)
114+
119115
args = parser.parse_args()
120116

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

136-
if args.quantized and "4bit-quantized" not in args.model_version:
137-
args.model_version += "-4bit-quantized"
138-
logger.info(f"Using 4-bit quantized model: {args.model_version}")
139-
140132
shift = args.shift or SHIFT[args.model_version]
141133
pipeline_class = FluxPipeline if "FLUX" in args.model_version else DiffusionPipeline
142134

0 commit comments

Comments
 (0)