alphard/stable-diffusion.cpp
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat: add sdxl v-pred suppport (#536)

Browse Source
This commit is contained in:
stduhpf 2025-01-18 06:15:54 +01:00 committed by GitHub
parent 587a37b2e2
commit d9b5942d98
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
stable-diffusion.cpp
View File

@ -525,6 +525,10 @@ public:
if (is_using_v_parameterization_for_sd2(ctx, sd_version_is_inpaint(version))) { if (is_using_v_parameterization_for_sd2(ctx, sd_version_is_inpaint(version))) {
is_using_v_parameterization = true; is_using_v_parameterization = true;
} }
} else if (sd_version_is_sdxl(version)) {
if (model_loader.tensor_storages_types.find("v_pred") != model_loader.tensor_storages_types.end()) {
is_using_v_parameterization = true;
}
} else if (version == VERSION_SVD) { } else if (version == VERSION_SVD) {
// TODO: V_PREDICTION_EDM // TODO: V_PREDICTION_EDM
is_using_v_parameterization = true; is_using_v_parameterization = true;