vall-e/vall_e/utils
2024-08-04 19:56:21 -05:00
..
__init__.py fix issue with sft and shared tensors... 2024-08-04 19:56:21 -05:00
distributed.py
io.py implicitly load either normal pickled weights or safetensors on loading the model 2024-08-03 23:34:18 -05:00
pattern.py
sampler.py
trainer.py
utils.py fix issue with sft and shared tensors... 2024-08-04 19:56:21 -05:00
wrapper.py changed torch.Tensor().to(device, dtype) to just torch.tensor(..., device, dtype) because it's been bothering my autism that I'm creating tensors then converting rather than creating with the right device/dtype, some 'optimization' to compile the model but it doesnt seem to do anything useful 2024-08-03 22:10:21 -05:00