Closed ggozad closed 1 year ago
torch.Generator is not yet available for MPS devices. Here's a small fix to fallback to cpu.
LGTM! Thanks so much for this...in back of my head I knew this was an issue but didn't have chance to fix it.
torch.Generator is not yet available for MPS devices. Here's a small fix to fallback to cpu.