Skip to content

Commit 225162f

Browse files
authored
fix: mark encoder.embed_tokens.weight as unused tensor (#721)
1 parent b9e4718 commit 225162f

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

model.cpp

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -100,6 +100,7 @@ const char* unused_tensors[] = {
100100
"model_ema.diffusion_model",
101101
"embedding_manager",
102102
"denoiser.sigmas",
103+
"text_encoders.t5xxl.transformer.encoder.embed_tokens.weight", // only used during training
103104
};
104105

105106
bool is_unused_tensor(std::string name) {

0 commit comments

Comments
 (0)