nodes/aigc-apps-EasyAnimate
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Update ui

Browse Source
This commit is contained in:
bubbliiiing
2025-01-22 06:04:26 +00:00
parent 81583f10f5
commit 1509f1c383
1 changed files with 4 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
easyanimate/ui/ui.py
+4 -2
View File
@@ -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(