Skip to content

Commit

Permalink
Make expandable segments on by default
Browse files Browse the repository at this point in the history
  • Loading branch information
b-chu authored Jun 13, 2024
1 parent ffec54b commit 86d821f
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion llmfoundry/utils/config_utils.py
Original file line number Diff line number Diff line change
@@ -115,7 +115,7 @@ class TrainConfig:

# Cuda allocation configuration
max_split_size_mb: Optional[int] = None
expandable_segments: bool = False
expandable_segments: bool = True
cuda_load_lazy: bool = False

# Distributed training parameters

0 comments on commit 86d821f

Please sign in to comment.