Pre-trained Style Encoder gives Nan values

[Conscht]

Issue: Using the extracted pre-trained style encoder results in output embeddings with NaN values, but only during training.

Resolution: Training was initially running on mixed precision at 16 bits. However, changing it to 32 bits resolved the issue, and only correct embeddings were generated.