Skip to content

Commit 9a269c4

Browse files
committed
feat: call path on pre model load
1 parent 2d130f5 commit 9a269c4

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

src/axolotl/loaders/patch_manager.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -53,6 +53,7 @@ def apply_pre_model_load_patches(self):
5353
self._apply_fsdp_patches()
5454
self._apply_adapter_patches()
5555
self._apply_flex_attention_patches()
56+
self._apply_sageattn_patches()
5657
self._apply_model_specific_patches()
5758
self._apply_fp8_patches()
5859
self._apply_flash_attention_peft_patches()

0 commit comments

Comments
 (0)