Skip to content

Commit 957aec1

Browse files
committed
Chroma: disable guidance by default
1 parent c19ea0f commit 957aec1

File tree

1 file changed

+16
-0
lines changed

1 file changed

+16
-0
lines changed

flux.hpp

Lines changed: 16 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1113,6 +1113,22 @@ namespace Flux {
11131113
c_concat = to_backend(c_concat);
11141114
}
11151115
if (flux_params.is_chroma) {
1116+
const char* SD_CHROMA_ENABLE_GUIDANCE = getenv("SD_CHROMA_ENABLE_GUIDANCE");
1117+
bool disable_guidance = true;
1118+
if (SD_CHROMA_ENABLE_GUIDANCE != NULL) {
1119+
std::string enable_guidance_str = SD_CHROMA_ENABLE_GUIDANCE;
1120+
if (enable_guidance_str == "ON" || enable_guidance_str == "TRUE") {
1121+
LOG_WARN("Chroma guidance has been enabled. Image might be broken. (SD_CHROMA_ENABLE_GUIDANCE env variable to \"OFF\" to disable)", SD_CHROMA_ENABLE_GUIDANCE);
1122+
disable_guidance = false;
1123+
} else if (enable_guidance_str != "OFF" && enable_guidance_str != "FALSE") {
1124+
LOG_WARN("SD_CHROMA_ENABLE_GUIDANCE environment variable has unexpected value. Assuming default (\"OFF\"). (Expected \"ON\"/\"TRUE\" or\"OFF\"/\"FALSE\", got \"%s\")", SD_CHROMA_ENABLE_GUIDANCE);
1125+
}
1126+
}
1127+
if (disable_guidance) {
1128+
LOG_DEBUG("Forcing guidance to 0 for chroma model (SD_CHROMA_ENABLE_GUIDANCE env variable to \"ON\" to enable)");
1129+
guidance = ggml_set_f32(guidance, 0);
1130+
}
1131+
11161132
int mask_pad = 1;
11171133
const char* SD_CHROMA_MASK_PAD_OVERRIDE = getenv("SD_CHROMA_MASK_PAD_OVERRIDE");
11181134
if (SD_CHROMA_MASK_PAD_OVERRIDE != nullptr) {

0 commit comments

Comments
 (0)