Merge branch 'main' into i18n-build-mode

2026-04-23 03:00:31 -04:00 · 2023-03-29 05:33:14 -07:00
parent 78e76f26f9 30ca25897e
commit e6123eac19
2 changed files with 4 additions and 5 deletions
--- a/invokeai/backend/model_management/convert_ckpt_to_diffusers.py
+++ b/invokeai/backend/model_management/convert_ckpt_to_diffusers.py
@@ -1264,10 +1264,10 @@ def load_pipeline_from_original_stable_diffusion_ckpt(
                cache_dir=cache_dir,
            )
            pipe = pipeline_class(
-                vae=vae,
-                text_encoder=text_model,
+                vae=vae.to(precision),
+                text_encoder=text_model.to(precision),
                tokenizer=tokenizer,
-                unet=unet,
+                unet=unet.to(precision),
                scheduler=scheduler,
                safety_checker=None,
                feature_extractor=None,