Skip to content

Commit 03c33eb

Browse files
committed
Chroma/Pixart T5: Disable attention masking by default
1 parent e767be7 commit 03c33eb

File tree

1 file changed

+5
-5
lines changed

1 file changed

+5
-5
lines changed

conditioner.hpp

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -1475,14 +1475,14 @@ struct PixArtCLIPEmbedder : public Conditioner {
14751475

14761476
size_t chunk_count = t5_tokens.size() / chunk_len;
14771477

1478-
bool use_mask = true;
1478+
bool use_mask = false;
14791479
const char* SD_CHROMA_USE_T5_MASK = getenv("SD_CHROMA_USE_T5_MASK");
14801480
if (SD_CHROMA_USE_T5_MASK != nullptr) {
14811481
std::string sd_chroma_use_t5_mask_str = SD_CHROMA_USE_T5_MASK;
1482-
if (sd_chroma_use_t5_mask_str == "OFF" || sd_chroma_use_t5_mask_str == "FALSE") {
1483-
use_mask = false;
1484-
} else if (sd_chroma_use_t5_mask_str != "ON" && sd_chroma_use_t5_mask_str != "TRUE") {
1485-
LOG_WARN("SD_CHROMA_USE_T5_MASK environment variable has unexpected value. Assuming default (\"ON\"). (Expected \"ON\"/\"TRUE\" or\"OFF\"/\"FALSE\", got \"%s\")", SD_CHROMA_USE_T5_MASK);
1482+
if (sd_chroma_use_t5_mask_str == "ON" || sd_chroma_use_t5_mask_str == "TRUE") {
1483+
use_mask = true;
1484+
} else if (sd_chroma_use_t5_mask_str != "OFF" && sd_chroma_use_t5_mask_str != "FALSE") {
1485+
LOG_WARN("SD_CHROMA_USE_T5_MASK environment variable has unexpected value. Assuming default (\"OFF\"). (Expected \"OFF\"/\"FALSE\" or\"ON\"/\"TRUE\", got \"%s\")", SD_CHROMA_USE_T5_MASK);
14861486
}
14871487
}
14881488
for (int chunk_idx = 0; chunk_idx < chunk_count; chunk_idx++) {

0 commit comments

Comments
 (0)