Closed rsxdalv closed 1 year ago
Did
load_from_checkpoint
on cpu not work? I thought using theto()
method would move both the model and the weights to the correct device. Either way, better to be safe than sorry.I see you also renamed the folder, great, I hope that projects which use it as a submodule will update then. Giving it an accurate name such as
bark_hubert_quantizer
instead ofhubert
is clearly the better name.And adding an install with setuptools instead of requirements.txt might also be great for some users.
That's the weird thing, I had to do both, just doing torch.load(...map_location) was not enough.
Did
load_from_checkpoint
on cpu not work? I thought using theto()
method would move both the model and the weights to the correct device. Either way, better to be safe than sorry.I see you also renamed the folder, great, I hope that projects which use it as a submodule will update then. Giving it an accurate name such as
bark_hubert_quantizer
instead ofhubert
is clearly the better name.And adding an install with setuptools instead of requirements.txt might also be great for some users.