hf-internal-testing

/

tiny-random-Swin2SRForImageSuperResolution

Like 0

Follow

Hugging Face Internal Testing Organization 180

# Use a pipeline as a high-level helper
# Warning: Pipeline type "image-to-image" is no longer supported in transformers v5.
# You must load the model directly (see below) or downgrade to v4.x with:
# pip install "transformers<5.0.0"
from transformers import pipeline

pipe = pipeline("image-to-image", model="hf-internal-testing/tiny-random-Swin2SRForImageSuperResolution")

# pip install -U transformers accelerate
# Load model directly
from transformers import AutoImageProcessor, AutoModelForImageToImage

processor = AutoImageProcessor.from_pretrained("hf-internal-testing/tiny-random-Swin2SRForImageSuperResolution")
model = AutoModelForImageToImage.from_pretrained("hf-internal-testing/tiny-random-Swin2SRForImageSuperResolution", device_map="auto")