feat: add PiD support (#1585)

This commit is contained in:
leejet authored and GitHub committed 2026-05-31 22:38:39 +08:00
1 parent d2797b8667
commit 0982807139
23 files changed
+1387 -23

No files matched your search

+10 -1
View File
@@ -49,6 +49,7 @@ enum SDVersion {
VERSION_ERNIE_IMAGE,
VERSION_LENS,
VERSION_LONGCAT,
VERSION_PID,
VERSION_COUNT,
};
@@ -164,6 +165,13 @@ static inline bool sd_version_is_lens(SDVersion version) {
return false;
}
static inline bool sd_version_is_pid(SDVersion version) {
if (version == VERSION_PID) {
return true;
}
return false;
}
static inline bool sd_version_uses_flux2_vae(SDVersion version) {
if (sd_version_is_flux2(version) || sd_version_is_ernie_image(version) || sd_version_is_lens(version)) {
return true;
@@ -194,7 +202,8 @@ static inline bool sd_version_is_dit(SDVersion version) {
sd_version_is_z_image(version) ||
sd_version_is_ernie_image(version) ||
sd_version_is_lens(version) ||
sd_version_is_longcat(version)) {
sd_version_is_longcat(version) ||
sd_version_is_pid(version)) {
return true;
}
return false;