cb9a1eef01
feature_utils.to(device) was loading CLIP ViT-H, synchformer, T5, VAE, and vocoder (~90 GiB) to GPU for the entire training run. Now only mel_converter (tiny) is moved to GPU. Pre-generation manages its own device placement: temporarily moves CLIP and tod to GPU, then moves them back when done. This frees ~90 GiB for the backward pass. Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>