mirror of
https://github.com/wassname/alignment-handbook.git
synced 2026-08-11 11:14:37 +08:00
Refactor imports
This commit is contained in:
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user