How to use lhallee/unet with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("image-segmentation", model="lhallee/unet")
# Load model directly from transformers import UNetForSegmentation model = UNetForSegmentation.from_pretrained("lhallee/unet", dtype="auto")
