Along with reformatting to pass black lint. - egs/libritts/ASR/zipformer/train.py - egs/libritts/CODEC/encodec/encodec.py - egs/ljspeech/TTS/vits/vits.py - egs/wenetspeech4tts/TTS/valle/train.py
with autocast(...)
with autocast("cuda", ...)