File tree 1 file changed +1
-9
lines changed
python/src/diffusionkit/mlx/scripts 1 file changed +1
-9
lines changed Original file line number Diff line number Diff line change @@ -111,11 +111,7 @@ def cli():
111
111
type = str ,
112
112
help = "Path to the local mmdit checkpoint." ,
113
113
)
114
- parser .add_argument (
115
- "--quantized" ,
116
- action = "store_true" ,
117
- help = "Use 4-bit quantized model if available" ,
118
- )
114
+
119
115
args = parser .parse_args ()
120
116
121
117
args .w16 = True
@@ -133,10 +129,6 @@ def cli():
133
129
if args .denoise < 0.0 or args .denoise > 1.0 :
134
130
raise ValueError ("Denoising factor must be between 0.0 and 1.0" )
135
131
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
-
140
132
shift = args .shift or SHIFT [args .model_version ]
141
133
pipeline_class = FluxPipeline if "FLUX" in args .model_version else DiffusionPipeline
142
134
You can’t perform that action at this time.
0 commit comments