mirror of
https://github.com/zenlm/zen-family.git
synced 2026-11-02 17:32:06 +00:00
25 lines
476 B
Python
25 lines
476 B
Python
"""
|
|
Configuration module for Zen models
|
|
"""
|
|
|
|
from .model_configs import (
|
|
MODEL_CONFIGS,
|
|
QLORA_CONFIGS,
|
|
DEEPSPEED_CONFIGS,
|
|
get_model_config,
|
|
get_qlora_config,
|
|
get_deepspeed_config,
|
|
list_available_models,
|
|
get_optimal_batch_size
|
|
)
|
|
|
|
__all__ = [
|
|
'MODEL_CONFIGS',
|
|
'QLORA_CONFIGS',
|
|
'DEEPSPEED_CONFIGS',
|
|
'get_model_config',
|
|
'get_qlora_config',
|
|
'get_deepspeed_config',
|
|
'list_available_models',
|
|
'get_optimal_batch_size'
|
|
] |