Compare commits

...

1 Commits

Author SHA1 Message Date
Dhruv Nair
251b1f1d6d update 2025-03-03 12:58:10 +01:00

View File

@@ -1448,8 +1448,8 @@ def convert_open_clip_checkpoint(
if text_proj_key in checkpoint:
text_proj_dim = int(checkpoint[text_proj_key].shape[0])
elif hasattr(text_model.config, "projection_dim"):
text_proj_dim = text_model.config.projection_dim
elif hasattr(text_model.config, "hidden_dim"):
text_proj_dim = text_model.config.hidden_dim
else:
text_proj_dim = LDM_OPEN_CLIP_TEXT_PROJECTION_DIM