feat: add krea2 support (#1705)

This commit is contained in:
leejet authored and GitHub committed 2026-06-25 00:25:50 +08:00
1 parent 2938272d82
commit 8caa3f908a
10 files changed
+803 -6

No files matched your search

+10 -1
View File
@@ -49,6 +49,7 @@ enum SDVersion {
VERSION_LONGCAT,
VERSION_PID,
VERSION_IDEOGRAM4,
VERSION_KREA2,
VERSION_ESRGAN,
VERSION_COUNT,
};
@@ -186,6 +187,13 @@ static inline bool sd_version_is_ideogram4(SDVersion version) {
return false;
}
static inline bool sd_version_is_krea2(SDVersion version) {
if (version == VERSION_KREA2) {
return true;
}
return false;
}
static inline bool sd_version_uses_flux_vae(SDVersion version) {
if (sd_version_is_flux(version) || sd_version_is_z_image(version) || sd_version_is_boogu_image(version) || sd_version_is_longcat(version)) {
return true;
@@ -226,7 +234,8 @@ static inline bool sd_version_is_dit(SDVersion version) {
sd_version_is_lens(version) ||
sd_version_is_longcat(version) ||
sd_version_is_pid(version) ||
sd_version_is_ideogram4(version)) {
sd_version_is_ideogram4(version) ||
sd_version_is_krea2(version)) {
return true;
}
return false;