AI/diffusers
1
0
Fork 0
mirror of https://github.com/huggingface/diffusers.git synced 2026-01-27 17:22:53 +03:00
Code Activity

fix loading custom text encoder when using from_single_file (#4571)

Browse Source 
fix loading custom text encoder when using from_single_file
This commit is contained in:
Dhruv Nair
2023-08-17 08:41:09 +05:30
committed by GitHub
parent bdc4c3265f
commit 3d08d8dc4e
1 changed files with 2 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/diffusers/pipelines/stable_diffusion/convert_from_ckpt.py
View File

@@ -783,6 +783,8 @@ def convert_ldm_clip_checkpoint(checkpoint, local_files_only=False, text_encoder
ctx = init_empty_weights if is_accelerate_available() else nullcontext
with ctx():
text_model = CLIPTextModel(config)
else:
text_model = text_encoder
keys = list(checkpoint.keys())