Refactor imports

This commit is contained in:
Lewis Tunstall
2023-11-10 13:38:45 +00:00
parent 7f1a14e0d4
commit e2e8ab945d
6 changed files with 58 additions and 18 deletions
+1 -1
View File
@@ -2,4 +2,4 @@ __version__ = "0.2.0.dev0"
from .configs import DataArguments, DPOConfig, H4ArgumentParser, ModelArguments, SFTConfig
from .data import apply_chat_template, get_datasets
from .model_utils import get_kbit_device_map, get_peft_config, get_quantization_config, get_tokenizer
from .model_utils import get_kbit_device_map, get_peft_config, get_quantization_config, get_tokenizer, is_adapter_model