PROSE-FD model weights

.gitattributes

@@ -33,3 +33,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zip filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
+*png filter=lfs diff=lfs merge=lfs -text

README.md

@@ -1,3 +1,22 @@
 ---
 license: mit
 ---
+
+# PROSE-FD: A Multimodal PDE Foundation Model for Learning Multiple Operators for Forecasting Fluid Dynamics
+
+This folder contains model weights for the PROSE-FD model in paper [PROSE-FD: A Multimodal PDE Foundation Model for Learning Multiple Operators for Forecasting Fluid Dynamics](https://arxiv.org/abs/2409.09811). The code is included in: https://github.com/felix-lyx/prose.
+
+![model](model.png)
+
+Note: The keys for the weights contain ```module._orig_mod.``` prefixes (coming from DDP and torch.compile). This is automatically taken care of in our code, but worth noting. 
+
+## Citation
+
+```
+@article{liu2024prose_fd,
+  title={{PROSE-FD}: A Multimodal PDE Foundation Model for Learning Multiple Operators for Forecasting Fluid Dynamics},
+  author={Liu, Yuxuan and Sun, Jingmin and He, Xinjie and Pinney, Griffin and Zhang, Zecheng and Schaeffer, Hayden},
+  journal={arXiv preprint arXiv:2409.09811},
+  year={2024}
+}
+```

prose_fd.pth

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:7fa0e65f9be922bbf99dfb5fd1fc0ac072fbcca99574d646db022f1399691bd1
+size 665395159