SIGN IN SIGN UP

Add `text_encoder_dtype` and `compile_text_encoder` config parameters for Wan text encoders.

Mitigates PR#397 by enabling the text encoder dtype to be configured separately from `weights_dtype`. Separating these parameters makes it possible to compile/load the text encoder in float32 (as it was before #397 by default) while keeping the model weights in bfloat16, and provides a configuration parameter to make the text encoder compilation optional  (as it wasn’t compiled before #397).

This addresses problems since some environments experience issues  when `torch.compile` is run on it in hermetic/packaged setups.
N
Nina Shvetsova committed
46e96dad411d1862fb95ac14b7b91045fb79f907
Parent: 19d4e4d