Add TorchScript cosmoformer model + README

README.md

@@ -0,0 +1,12 @@
+# CosmoFormer Model
+
+This is a TorchScript version of our CrossFormer-based model.
+You can load it in plain PyTorch without needing `vit_pytorch`.
+
+Usage Example:
+```python
+import torch
+model = torch.jit.load('cosmoformer_traced.pt')
+model.eval()
+...
+```

cosmoformer_traced.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:25c9769ae5193f67e357facbde4fc956263ef2287a093437a918bb4a9879f9db
+size 23450310