Hamid Shojanazeri 44ef280d31 adding flash attention and xformer memory efficient through PT SDPA 1 år sedan
..
__init__.py 4767f09ecd Initial commit 1 år sedan
config_utils.py 4767f09ecd Initial commit 1 år sedan
dataset_utils.py 4767f09ecd Initial commit 1 år sedan
fsdp_utils.py 4767f09ecd Initial commit 1 år sedan
memory_utils.py 41dd7ff1cb Merge branch 'main' into checkpoint_handler_path_fix 1 år sedan
train_utils.py 44ef280d31 adding flash attention and xformer memory efficient through PT SDPA 1 år sedan