Constants
gllm_training.constants
Contains all constants used throughout the gllm_training package.
Reviewer
- Muhammad Afif Al Hawari (muhammad.a.a.hawari@gdplabs.id)
References
NONE
GPU_AVAILABLE = torch.cuda.is_available()
module-attribute
GeneralConstants(TEMPLATE_TRUNCATE_LENGTH=100, MAX_FAILED_ROWS_TO_LOG=5, RANDOM_SEED=42, CACHE_TIMEOUT=300, HYPERPARAMETER_PREFIX='fine_tune_hyperparam_', EXPERIMENT_DIR_TEMPLATE='{base_output_dir}/exp_{experiment_id}/{model_name}')
dataclass
Constants for general configuration.
CACHE_TIMEOUT: int = 300
class-attribute
instance-attribute
EXPERIMENT_DIR_TEMPLATE: str = '{base_output_dir}/exp_{experiment_id}/{model_name}'
class-attribute
instance-attribute
HYPERPARAMETER_PREFIX: str = 'fine_tune_hyperparam_'
class-attribute
instance-attribute
MAX_FAILED_ROWS_TO_LOG: int = 5
class-attribute
instance-attribute
RANDOM_SEED: int = 42
class-attribute
instance-attribute
TEMPLATE_TRUNCATE_LENGTH: int = 100
class-attribute
instance-attribute
PromptTemplateConstants(SYSTEM_PROMPT='You are a helpful assistant.', USER_PROMPT='{query}', SYSTEM_PROMPT_COL='system', USER_PROMPT_COL='user', PROMPT_ID_COL='name')
dataclass
Constants for prompt template processing.