Flizyx opened this issue a year ago · 1 comments
It is running on the GPU, check for this line:
model.to(torch_device="cpu", torch_dtype=torch.float32).to('mps:0')
If I'm correct, that is telling the machine to use MPS (Metal Performance Shaders)
Those three spikes in my screenshot are my tests: