Fix Ovis image text encoder dtype (#3876)

Signed-off-by: akshatvishu <akshatnayak197@gmail.com>
This commit is contained in:
akshatvishu authored and GitHub committed 2026-05-28 08:21:37 +08:00
1 parent d37912b071
commit 1f3a3e3848
1 file changed
+4 -1
@@ -168,7 +168,10 @@ class OvisImagePipeline(nn.Module, CFGParallelMixin, DiffusionPipelineProfilerMi
)
self.text_encoder = Qwen3Model.from_pretrained(
model, subfolder="text_encoder", local_files_only=local_files_only
model,
subfolder="text_encoder",
local_files_only=local_files_only,
torch_dtype=od_config.dtype,
)
self.vae = AutoencoderKL.from_pretrained(model, subfolder="vae", local_files_only=local_files_only).to(