upstream-archive/stable-diffusion.cpp
1
0
Fork 0
mirror of https://github.com/leejet/stable-diffusion.cpp.git synced 2026-07-23 11:20:53 -05:00
Code Issues Packages Projects Releases Wiki Activity

refactor: add some sd vesion helper functions

Browse Source
This commit is contained in:
leejet
2024-11-23 13:02:44 +08:00
parent 1c168d98a5
commit b5f4932696
5 changed files with 59 additions and 38 deletions
Download Patch File Download Diff File Expand all files Collapse all files

25
model.h
View File

@@ -31,9 +31,30 @@ enum SDVersion {
VERSION_COUNT,
};
static inline bool sd_version_is_flux(SDVersion version) {
if (version == VERSION_FLUX_DEV || version == VERSION_FLUX_SCHNELL || version == VERSION_FLUX_LITE) {
return true;
}
return false;
}
static inline bool sd_version_is_sd3(SDVersion version) {
if (version == VERSION_SD3_2B || version == VERSION_SD3_5_8B || version == VERSION_SD3_5_2B) {
return true;
}
return false;
}
static inline bool sd_version_is_dit(SDVersion version) {
if (sd_version_is_flux(version) || sd_version_is_sd3(version)) {
return true;
}
return false;
}
enum PMVersion {
VERSION_1,
VERSION_2,
PM_VERSION_1,
PM_VERSION_2,
};
struct TensorStorage {