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