Hugging Face's logo

tangledgroup
/
tangled-llama-j-128k-v0.1

llama
Model card Files Community
mtasic85 commited on
Commit
3ff731f
·
1 Parent(s): 7caef11

pretrain

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -28,5 +28,5 @@ python -B prepare_pretrain_dataset.py
28
  ```
29
 
30
  ```bash
31
- CUDA_LAUNCH_BLOCKING=0 CUDA_VISIBLE_DEVICES="0" litgpt pretrain --config pretrain-model.yaml
32
  ```
 
28
  ```
29
 
30
  ```bash
31
+ CUDA_VISIBLE_DEVICES=0 CUDA_LAUNCH_BLOCKING=0 PYTORCH_CUDA_ALLOC_CONF=expandable_segments:True litgpt pretrain --config pretrain-model.yaml
32
  ```