tocmo0nlord/axolotl
1
0
Fork 0
Code Issues Pull Requests Actions 3 Packages Projects Releases Wiki Activity

stuff

Browse Source
This commit is contained in:
sunny
2024-10-30 13:44:06 -04:00
parent 38773d661f
commit bfb80a3ef9
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/axolotl/utils/models.py
View File

@@ -393,7 +393,7 @@ class ModelLoader:
self.cfg.model_config_type in SUPPORTED_MULTIPACK_MODEL_TYPES self.cfg.model_config_type in SUPPORTED_MULTIPACK_MODEL_TYPES
and self.cfg.flash_attention and self.cfg.flash_attention
and self.cfg.sample_packing and self.cfg.sample_packing
): ):
has_remote_code = ( has_remote_code = (
"auto_map" in self.model_config "auto_map" in self.model_config
and self.model_type in self.model_config["auto_map"] and self.model_type in self.model_config["auto_map"]