support Apple mps

README.md

@@ -45,6 +45,7 @@ from hit_sir_arch import HiT_SIR
 from hit_sng_arch import HiT_SNG
 from hit_srf_arch import HiT_SRF
 import cv2
+import torch
 
 # detect device
 device = torch.device('cuda' if torch.cuda.is_available() else 'mps' if torch.backends.mps.is_available() else 'cpu')