InternLM/internlm/initialize
ytxiong 307c4741d1
fix(initialize/launch.py): set default value for use_flash_attn (#158)
* add default for use_flash_attn

* fix lint
2023-08-01 16:03:06 +08:00
..
__init__.py initial commit 2023-07-06 12:55:23 +08:00
initialize_tensor.py initial commit 2023-07-06 12:55:23 +08:00
initialize_trainer.py feat(*): support not-flash-attn for pp and no-pp (#145) 2023-07-28 16:13:04 +08:00
launch.py fix(initialize/launch.py): set default value for use_flash_attn (#158) 2023-08-01 16:03:06 +08:00