dc-ae and text-encoder load

[wanglaofei]

How to config the local DC-AE and Text-Encoder path? I have downloaded the related models in my local computer, but I don't know how to load the local DC-AE and Text-Encoder. Thanks for any help!

[moeinheidari7829]

How to config the local DC-AE and Text-Encoder path? I have downloaded the related models in my local computer, but I don't know how to load the local DC-AE and Text-Encoder. Thanks for any help!

Hi. I hope you are doing well. Can you please tell me how you manually downloaded the checkpoints to your local?

[lawrence-cj]

For vae and text encoder loading:

Sana/diffusion/model/builder.py

Line 80 in a8f3488

def get_vae(name, model_path, device="cuda"):

Sana/diffusion/model/builder.py

Line 44 in a8f3488

def get_tokenizer_and_text_encoder(name="T5", device="cuda"):