Ensure you place the SAM checkpoint here to avoid the need to alter the imports.