[From Single File] Allow Text Encoder to be passed (#6020)

Allow text encoder to be passed

[From Single File] Allow Text Encoder to be passed (#6020)
Allow text encoder to be passed
bc1d28c8 · Patrick von Platen · GitHub · af378c1d · bc1d28c8
Unverified Commit bc1d28c8 authored Dec 01, 2023 by Patrick von Platen Committed by GitHub Dec 01, 2023
Hide whitespace changes
Inline Side-by-side

Showing with 6 additions and 3 deletions

src/diffusers/pipelines/stable_diffusion/convert_from_ckpt.py ...diffusers/pipelines/stable_diffusion/convert_from_ckpt.py +6 -3

No files found.
--- a/src/diffusers/pipelines/stable_diffusion/convert_from_ckpt.py
+++ b/src/diffusers/pipelines/stable_diffusion/convert_from_ckpt.py
@@ -1480,9 +1480,12 @@ def download_from_original_stable_diffusion_ckpt(
        config_name = "stabilityai/stable-diffusion-2"
        config_kwargs = {"subfolder": "text_encoder"}
-        text_model = convert_open_clip_checkpoint(
+        if text_encoder is None:
-            checkpoint, config_name, local_files_only=local_files_only, **config_kwargs
+            text_model = convert_open_clip_checkpoint(
-        )
+                checkpoint, config_name, local_files_only=local_files_only, **config_kwargs
+            )
+        else:
+            text_model = text_encoder
        try:
            tokenizer = CLIPTokenizer.from_pretrained(