YongZJNZ Yong99 commited on
Commit
8911430
·
1 Parent(s): fe6dcc4

Update README.md (#1)

Browse files

- Update README.md (6a59992785a526ce74cc9875307dea8faee9001a)


Co-authored-by: Yong Liu <Yong99@users.noreply.huggingface.co>

Files changed (1) hide show
  1. README.md +3 -1
README.md CHANGED
@@ -88,7 +88,9 @@ print(output[0][4])
88
  ```
89
 
90
 
91
- > This model support inference using either CPU or GPU. To load this model on GPU, we recommend a GPU with **at least 40GB VRAM** (e.g., A100 40GB/80GB, or H100). **Encounter out-of-memory at runtime?** Try the following options:
 
 
92
  > ```python
93
  > # Option 1: reduce batch size or context length
94
  > batch_size, lookback_length = 1, 2880
 
88
  ```
89
 
90
 
91
+ This model support inference using either CPU or GPU. To load this model on GPU, we recommend a GPU with **at least 40GB VRAM** (e.g., A100 40GB/80GB, or H100).
92
+
93
+ > **Encounter out-of-memory at runtime?** Try the following options:
94
  > ```python
95
  > # Option 1: reduce batch size or context length
96
  > batch_size, lookback_length = 1, 2880