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
|