shank
Pin torch to cu121 build + use model.device instead of hardcoded cuda string
8f291e0