fix: fix the issue in parsing --chroma-disable-dit-mask

leejet · leejet · commit a28d04dd8190 · 2025-06-29T23:52:36.000+08:00
diff --git a/examples/cli/main.cpp b/examples/cli/main.cpp
@@ -653,7 +653,7 @@ void parse_args(int argc, const char** argv, SDParams& params) {
                 break;
             }
             params.ref_image_paths.push_back(argv[i]);
-        } else if (arg == "chroma-disable-dit-mask") {
+        } else if (arg == "--chroma-disable-dit-mask") {
             params.chroma_use_dit_mask = false;
         } else if (arg == "--chroma-use-t5-mask") {
             params.chroma_use_t5_mask = true;

Original file line number	Diff line number	Diff line change
`@@ -653,7 +653,7 @@ void parse_args(int argc, const char** argv, SDParams& params) {`
`653`	`653`	`break;`
`654`	`654`	`}`
`655`	`655`	`params.ref_image_paths.push_back(argv[i]);`
`656`		`- } else if (arg == "chroma-disable-dit-mask") {`
	`656`	`+ } else if (arg == "--chroma-disable-dit-mask") {`
`657`	`657`	`params.chroma_use_dit_mask = false;`
`658`	`658`	`} else if (arg == "--chroma-use-t5-mask") {`
`659`	`659`	`params.chroma_use_t5_mask = true;`