PyTorch 2.0

#2
by Denny-KEF - opened

Is it possible to use torch.compile(model) with this implementation? I load the model, then use torch to load my trained weights and would like to use torch 2.0 to improve the speed but I am having some issues. I am wondering if anyone has gotten it to work?

I actually made a PR and fixed the issue.

Denny-KEF changed discussion status to closed

Sign up or log in to comment