how to enable GPU on m1 ?

[Hans-han]

Hi, Since PyTorch now support MPS acceleration I wonder how would I enable gpu using GPU parameters.

import torch
device = torch.device("mps")

reader = easyocr.Reader(['ch_sim',"en"],gpu=device)

above code does not work