Skip to content

Commit 05e428c

Browse files
committed
SDXL v-pred
1 parent 9578fdc commit 05e428c

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

stable-diffusion.cpp

+4
Original file line numberDiff line numberDiff line change
@@ -521,6 +521,10 @@ class StableDiffusionGGML {
521521
if (is_using_v_parameterization_for_sd2(ctx)) {
522522
is_using_v_parameterization = true;
523523
}
524+
} else if (sd_version_is_sdxl(version)) {
525+
if (model_loader.tensor_storages_types.find("v_pred") != model_loader.tensor_storages_types.end()) {
526+
is_using_v_parameterization = true;
527+
}
524528
} else if (version == VERSION_SVD) {
525529
// TODO: V_PREDICTION_EDM
526530
is_using_v_parameterization = true;

0 commit comments

Comments
 (0)