Use model instead of image size to detect the A100