How to use phungpx/mask2former-swin-base-cubicassa with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("image-segmentation", model="phungpx/mask2former-swin-base-cubicassa")
# Load model directly from transformers import AutoImageProcessor, Mask2FormerForUniversalSegmentation processor = AutoImageProcessor.from_pretrained("phungpx/mask2former-swin-base-cubicassa") model = Mask2FormerForUniversalSegmentation.from_pretrained("phungpx/mask2former-swin-base-cubicassa", device_map="auto")
