vall-e/vall_e/engines
2024-04-09 14:41:13 -05:00
..
__init__.py actually do the Linear replacement with TE's Linear 2024-04-09 14:41:13 -05:00
base.py added FP8 support through NVIDIA/TransformerEngine, added RetNet_HF through syncdoth/RetNet (as an alternative to branch away from torchscale) 2024-04-08 20:14:51 -05:00
deepspeed.py added FP8 support through NVIDIA/TransformerEngine, added RetNet_HF through syncdoth/RetNet (as an alternative to branch away from torchscale) 2024-04-08 20:14:51 -05:00