feat: support for --tensor-type-rules on generation modes (#932)

This commit is contained in:
Wagner Bruna
2025-11-16 06:07:32 -03:00
committed by GitHub
parent 742a7333c3
commit 199e675cc7
5 changed files with 59 additions and 49 deletions

View File

@@ -1241,10 +1241,6 @@ void parse_args(int argc, const char** argv, SDParams& params) {
exit(1);
}
if (params.mode != CONVERT && params.tensor_type_rules.size() > 0) {
fprintf(stderr, "warning: --tensor-type-rules is currently supported only for conversion\n");
}
if (params.mode == VID_GEN && params.video_frames <= 0) {
fprintf(stderr, "warning: --video-frames must be at least 1\n");
exit(1);
@@ -1756,6 +1752,7 @@ int main(int argc, const char* argv[]) {
params.lora_model_dir.c_str(),
params.embedding_dir.c_str(),
params.photo_maker_path.c_str(),
params.tensor_type_rules.c_str(),
vae_decode_only,
true,
params.n_threads,