Spaces:

Dovakiins
/

qwerrwe

Build error

App Files Files Community

qwerrwe / src /axolotl /utils

98.1 kB

100 contributors

History: 238 commits

winglian's picture

use context manager to run things on rank0 before others (#397)

fc2d6be unverified over 2 years ago

__init__.py

0 Bytes

WIP large refactor to make finetune script a little more manageable (#3) over 2 years ago
bench.py

1.24 kB

Fix crash when running without CUDA over 2 years ago
callbacks.py

2.75 kB

improve GPU logging to break out pytorch cache and system mem over 2 years ago
collators.py

5.39 kB

Attention mask and position id fixes for packing (#285) over 2 years ago
config.py

7.49 kB

try to detect accelerate and only use device_map=None in that case (#373) over 2 years ago
data.py

21.7 kB

use context manager to run things on rank0 before others (#397) over 2 years ago
dataloader.py

9.43 kB

Attention mask and position id fixes for packing (#285) over 2 years ago
dict.py

326 Bytes

fix DefaultDict.__or__ over 2 years ago
distributed.py

1.31 kB

use context manager to run things on rank0 before others (#397) over 2 years ago
models.py

19.3 kB

don't use mask expansion for inference (#392) over 2 years ago
schedulers.py

3.58 kB

add axolotl trainer and quadratic warmup over 2 years ago
tokenization.py

1.37 kB

Adding logging enhancement over 2 years ago
trainer.py

23.3 kB

Feat(config): add max steps (#387) over 2 years ago
wandb.py

858 Bytes

Add wandb_entity to wandb options, update example configs, update README (#361) over 2 years ago