Skip to content

Commit cd0b599

Browse files
committed
use pad token when padding sdxl chunks
1 parent 7b28226 commit cd0b599

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

conditioner.hpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -615,7 +615,7 @@ struct FrozenCLIPEmbedderWithCustomWords : public Conditioner {
615615
if (sd_version_is_sdxl(version)) {
616616
auto it = std::find(chunk_tokens.begin(), chunk_tokens.end(), tokenizer.EOS_TOKEN_ID);
617617
if (it != chunk_tokens.end()) {
618-
std::fill(std::next(it), chunk_tokens.end(), 0);
618+
std::fill(std::next(it), chunk_tokens.end(), tokenizer.PAD_TOKEN_ID);
619619
}
620620

621621
max_token_idx = std::min<size_t>(std::distance(chunk_tokens.begin(), it), chunk_tokens.size() - 1);

0 commit comments

Comments
 (0)