Skip to content

Commit bc1d28c

Browse files
[From Single File] Allow Text Encoder to be passed (#6020)
Allow text encoder to be passed
1 parent af378c1 commit bc1d28c

File tree

1 file changed

+6
-3
lines changed

1 file changed

+6
-3
lines changed

src/diffusers/pipelines/stable_diffusion/convert_from_ckpt.py

Lines changed: 6 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1480,9 +1480,12 @@ def download_from_original_stable_diffusion_ckpt(
14801480
config_name = "stabilityai/stable-diffusion-2"
14811481
config_kwargs = {"subfolder": "text_encoder"}
14821482

1483-
text_model = convert_open_clip_checkpoint(
1484-
checkpoint, config_name, local_files_only=local_files_only, **config_kwargs
1485-
)
1483+
if text_encoder is None:
1484+
text_model = convert_open_clip_checkpoint(
1485+
checkpoint, config_name, local_files_only=local_files_only, **config_kwargs
1486+
)
1487+
else:
1488+
text_model = text_encoder
14861489

14871490
try:
14881491
tokenizer = CLIPTokenizer.from_pretrained(

0 commit comments

Comments
 (0)