Transformers How to use camlab-ethz/Poseidon-L with Transformers:
# pip install -U transformers accelerate
# Load model directly
from transformers import AutoImageProcessor, SwinForPDE
processor = AutoImageProcessor.from_pretrained("camlab-ethz/Poseidon-L")
model = SwinForPDE.from_pretrained("camlab-ethz/Poseidon-L", device_map="auto") 
