From 1509f1c3833af848d981f2c79170ce1da1e27ba5 Mon Sep 17 00:00:00 2001 From: bubbliiiing <3323290568@qq.com> Date: Wed, 22 Jan 2025 06:04:26 +0000 Subject: [PATCH] Update ui --- easyanimate/ui/ui.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/easyanimate/ui/ui.py b/easyanimate/ui/ui.py index dc925e5..358562c 100644 --- a/easyanimate/ui/ui.py +++ b/easyanimate/ui/ui.py @@ -216,7 +216,8 @@ class EasyAnimateController: ) if self.inference_config['text_encoder_kwargs'].get('replace_t5_to_llm', False): text_encoder_2 = Qwen2VLForConditionalGeneration.from_pretrained( - os.path.join(diffusion_transformer_dropdown, "text_encoder_2") + os.path.join(diffusion_transformer_dropdown, "text_encoder_2"), + torch_dtype=self.weight_dtype, ) else: text_encoder_2 = T5EncoderModel.from_pretrained( @@ -225,7 +226,8 @@ class EasyAnimateController: else: if self.inference_config['text_encoder_kwargs'].get('replace_t5_to_llm', False): text_encoder = Qwen2VLForConditionalGeneration.from_pretrained( - os.path.join(diffusion_transformer_dropdown, "text_encoder") + os.path.join(diffusion_transformer_dropdown, "text_encoder"), + torch_dtype=self.weight_dtype, ) else: text_encoder = T5EncoderModel.from_pretrained(