feat: add generic DiT support to spectrum cache (#1336)

This commit is contained in:
rmatif
2026-03-15 09:41:05 +01:00
committed by GitHub
parent 6fa7ca9317
commit adfef62900
4 changed files with 6 additions and 6 deletions

View File

@@ -1797,9 +1797,9 @@ public:
}
}
} else if (cache_params->mode == SD_CACHE_SPECTRUM) {
bool spectrum_supported = sd_version_is_unet(version);
bool spectrum_supported = sd_version_is_unet(version) || sd_version_is_dit(version);
if (!spectrum_supported) {
LOG_WARN("Spectrum requested but not supported for this model type (only UNET models)");
LOG_WARN("Spectrum requested but not supported for this model type (only UNET and DiT models)");
} else {
SpectrumConfig spectrum_config;
spectrum_config.w = cache_params->spectrum_w;