fix: mark encoder.embed_tokens.weight as unused tensor (#721)

This commit is contained in:
idostyle 2025-07-06 17:10:10 +02:00 committed by GitHub
parent b9e4718fac
commit 225162f270
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -100,6 +100,7 @@ const char* unused_tensors[] = {
"model_ema.diffusion_model",
"embedding_manager",
"denoiser.sigmas",
"text_encoders.t5xxl.transformer.encoder.embed_tokens.weight", // only used during training
};
bool is_unused_tensor(std::string name) {