How to use openmmlab/upernet-convnext-base with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("image-segmentation", model="openmmlab/upernet-convnext-base")
# pip install -U transformers accelerate # Load model directly from transformers import AutoImageProcessor, UperNetForSemanticSegmentation processor = AutoImageProcessor.from_pretrained("openmmlab/upernet-convnext-base") model = UperNetForSemanticSegmentation.from_pretrained("openmmlab/upernet-convnext-base", device_map="auto")
