vall-e/vall_e/models
2023-10-13 18:22:21 -05:00
..
__init__.py tweaks to try and get deepspeed quantized inferencing, validating bitsandbytes and deepspeed quantization, nothing seems to work 2023-10-12 22:21:43 -05:00
adaln.py Tweaks 2023-08-02 22:06:39 +00:00
ar_nar.py mucked around with the loss calculation, this seems better? 2023-10-13 18:22:21 -05:00
ar.py actually use langs from the dataloader 2023-10-11 21:21:50 -05:00
base.py mucked around with the loss calculation, this seems better? 2023-10-13 18:22:21 -05:00
nar.py actually use langs from the dataloader 2023-10-11 21:21:50 -05:00
retnet.py restructured some things with the model to remove dead weights 2023-09-20 19:10:59 -05:00
transformer.py added ability to disable activation checkpointing through the YAML (it is very VRAM intensive at double layer size) 2023-09-05 15:38:21 -05:00