Update README.md
Browse files
README.md
CHANGED
@@ -95,7 +95,7 @@ python convert_t5x_checkpoint_to_pytorch.py --t5x_checkpoint_path ~/code/ul2/fla
|
|
95 |
|
96 |
## Running the model
|
97 |
|
98 |
-
For more efficient memory usage, we advise you to load the model in `8bit` using `load_in_8bit` flag as follows:
|
99 |
|
100 |
```python
|
101 |
# pip install accelerate transformers bitsandbytes
|
|
|
95 |
|
96 |
## Running the model
|
97 |
|
98 |
+
For more efficient memory usage, we advise you to load the model in `8bit` using `load_in_8bit` flag as follows (works only under GPU):
|
99 |
|
100 |
```python
|
101 |
# pip install accelerate transformers bitsandbytes
|