add suggested gpu requirements

README.md

@@ -24,6 +24,7 @@ See below for an [overview](#model) of the model and training. Better yet, check
 - [medium-450k](https://huggingface.co/LongSafari/hyenadna-medium-450k-seqlen/tree/main)
 - [large-1m](https://huggingface.co/LongSafari/hyenadna-large-1m-seqlen/tree/main)
 
+See [GPU requirements](#hardware) for each model.
 
 ### Sample snippet
 
@@ -82,6 +83,25 @@ Otherwise, checkout of the main HyenaDNA repo for how to load weights into Pytor
 If you want a standalone version that's easy to port into your own code (and not tied to our repo or Pytorch Lightning), we have that and a HuggingFace example in ['huggingface.py'](https://github.com/HazyResearch/hyena-dna/blob/main/huggingface.py) too.
 
 
+### GPU requirements (suggested)
+<a name="hardware"></a>
+
+Here are suggestions on the hardware (preferred minimum) we think you can use for each model.
+
+GPU during: Pretrain, fine-tune, inference
+
+- [tiny-1k](https://huggingface.co/LongSafari/hyenadna-tiny-1k-seqlen/tree/main): (T4, T4, T4)
+- [small-32k](https://huggingface.co/LongSafari/hyenadna-small-32k-seqlen/tree/main): (A100-40, T4, T4)
+- [medium-160k](https://huggingface.co/LongSafari/hyenadna-medium-160k-seqlen/tree/main): (A100-40, A100-40, T4)
+- [medium-450k](https://huggingface.co/LongSafari/hyenadna-medium-450k-seqlen/tree/main): (A100-40, A100-40, T4)
+- [large-1m](https://huggingface.co/LongSafari/hyenadna-large-1m-seqlen/tree/main): (A100-80, A100-80, A100-40)
+
+
+T4: 16GB  
+A100-40: 40GB  
+A100-80: 80GB  
+
+
 ## Model & Training Overview
 <a name="model"></a>
 
@@ -100,7 +120,7 @@ Check out our [blog](https://hazyresearch.stanford.edu/blog/2023-06-29-hyena-dna
 
 ### Authors
 
-Eric Nguyen*, Michael Poli*, Marjan Faizi*, Armin Thomas, Callum Birch-Sykes, Michael Wornow, Aman Patel, Stefano Massaroli, Clayton Rabideau, Yoshua Bengio, Stefano Ermon, Stephen Baccus, Chris Re.
+Eric Nguyen*, Michael Poli*, Marjan Faizi*, Armin Thomas, Callum Birch-Sykes, Michael Wornow, Aman Patel, Clayton Rabideau, Stefano Massaroli, Yoshua Bengio, Stefano Ermon, Stephen Baccus, Chris Re.
 
 **Contact**