Hamid Shojanazeri
44ef280d31
adding flash attention and xformer memory efficient through PT SDPA
|
1 年間 前 | |
|---|---|---|
| .. | ||
| __init__.py | 1 年間 前 | |
| config_utils.py | 1 年間 前 | |
| dataset_utils.py | 1 年間 前 | |
| fsdp_utils.py | 1 年間 前 | |
| memory_utils.py | 1 年間 前 | |
| train_utils.py | 1 年間 前 | |