Modify the code here to original_sizes = original_sizes.cpu().numpy() [sam](https://github.com/huggingface/transformers/blob/main/src/transformers/models/sam/processing_sam.py#L106)