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