Fix AIMV2 to use get_device() for MPS support#9

rromanchuk wants to merge 1 commit into

jacobmarks:mainjacobmarks/zero-shot-prediction-plugin:mainfrom

rromanchuk:fix/aimv2-mps-supportrromanchuk/zero-shot-prediction-plugin:fix/aimv2-mps-supportCopy head branch name to clipboard