How to use nielsr/vitmatte-small-composition-1k with Transformers:
# pip install -U transformers accelerate # Load model directly from transformers import AutoImageProcessor, VitMatteForImageMatting processor = AutoImageProcessor.from_pretrained("nielsr/vitmatte-small-composition-1k") model = VitMatteForImageMatting.from_pretrained("nielsr/vitmatte-small-composition-1k", device_map="auto")
SFconvertbot