Update README.md
Browse files
README.md
CHANGED
|
@@ -109,7 +109,4 @@ trainer = SFTTrainer(
|
|
| 109 |
),
|
| 110 |
)
|
| 111 |
|
| 112 |
-
# Set environment variables for PyTorch memory management
|
| 113 |
-
import os
|
| 114 |
-
os.environ["PYTORCH_CUDA_ALLOC_CONF"] = "max_split_size_mb:128,expandable_segments:True"
|
| 115 |
```
|
|
|
|
| 109 |
),
|
| 110 |
)
|
| 111 |
|
|
|
|
|
|
|
|
|
|
| 112 |
```
|