camlab-ethz/NS-Sines
Updated • 203
How to use camlab-ethz/Poseidon-T with Transformers:
# Load model directly
from transformers import AutoImageProcessor, SwinForPDE
processor = AutoImageProcessor.from_pretrained("camlab-ethz/Poseidon-T")
model = SwinForPDE.from_pretrained("camlab-ethz/Poseidon-T")The Tiny model of the Poseidon family with approx. 21M parameters, see https://arxiv.org/abs/2405.19101.
It is trained on a computational grid of size 128 x 128 and expects density, velocity in horizontal direction, velocity in vertical direction, and pressure as inputs.
This is a research model.