Update README.md
Browse files
README.md
CHANGED
|
@@ -66,4 +66,17 @@ loader.load_normalizer()
|
|
| 66 |
device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
|
| 67 |
model = model.to(device)
|
| 68 |
model.eval()
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 69 |
```
|
|
|
|
| 66 |
device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
|
| 67 |
model = model.to(device)
|
| 68 |
model.eval()
|
| 69 |
+
```
|
| 70 |
+
|
| 71 |
+
## Citation
|
| 72 |
+
|
| 73 |
+
If you use Suiren models, please cite the relevant papers for the underlying models.
|
| 74 |
+
|
| 75 |
+
```
|
| 76 |
+
@article{an2026suiren,
|
| 77 |
+
title={Suiren-1.0 Technical Report: A Family of Molecular Foundation Models},
|
| 78 |
+
author={An, Junyi and Lu, Xinyu and Shi, Yun-Fei and Xu, Li-Cheng and Zhang, Nannan and Qu, Chao and Qi, Yuan and Cao, Fenglei},
|
| 79 |
+
journal={arXiv preprint arXiv:2603.21942},
|
| 80 |
+
year={2026}
|
| 81 |
+
}
|
| 82 |
```
|