Skip to content

Commit

Permalink
Update on "remove old torch dependency in requirements.txt"
Browse files Browse the repository at this point in the history
`torch 2.2.0.dev` is too stale to be useful.
For CI, since we will install nightly anyway, this avoids storing the old version in the docker image.


[ghstack-poisoned]
  • Loading branch information
tianyu-l committed May 29, 2024
1 parent 31f53fe commit 593448a
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion train_configs/debug_model.toml
Original file line number Diff line number Diff line change
Expand Up @@ -3,6 +3,7 @@
[job]
dump_folder = "./outputs"
description = "Llama 3 debug training"
# TODO: turn this back on once ci have tokenizer
use_for_integration_test = true

[profiling]
Expand Down Expand Up @@ -49,7 +50,7 @@ interval_type = "steps"
interval = 5
model_weights_only = false
export_dtype = "float32"
async_mode = "disabled" # ["disabled", "async", "async_with_pinned_mem"]
async_mode = "disabled" # ["disabled", "async", "async_with_pinned_mem"]

[activation_checkpoint]
mode = 'selective' # ['none', 'selective', 'full']
Expand Down

0 comments on commit 593448a

Please sign in to comment.