Change checkpoint key used to identify CLIP models in single file checkpoints (#8319)

update

Change checkpoint key used to identify CLIP models in single file checkpoints (#8319)
update
f9fa8a86 · Dhruv Nair · GitHub · 05be622b · f9fa8a86
Unverified Commit f9fa8a86 authored May 31, 2024 by Dhruv Nair Committed by GitHub May 31, 2024
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

src/diffusers/loaders/single_file_utils.py src/diffusers/loaders/single_file_utils.py +1 -1

No files found.
--- a/src/diffusers/loaders/single_file_utils.py
+++ b/src/diffusers/loaders/single_file_utils.py
@@ -63,7 +63,7 @@ CHECKPOINT_KEY_NAMES = {
    "controlnet": "control_model.time_embed.0.weight",
    "playground-v2-5": "edm_mean",
    "inpainting": "model.diffusion_model.input_blocks.0.0.weight",
-    "clip": "cond_stage_model.transformer.text_model.embeddings.position_ids",
+    "clip": "cond_stage_model.transformer.text_model.embeddings.position_embedding.weight",
    "clip_sdxl": "conditioner.embedders.0.transformer.text_model.embeddings.position_embedding.weight",
    "open_clip": "cond_stage_model.model.token_embedding.weight",
    "open_clip_sdxl": "conditioner.embedders.1.model.positional_embedding",