Spaces:
Running
Running
File size: 1,553 Bytes
d8dd7a1 769bb84 d8dd7a1 769bb84 d8dd7a1 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 |
"""
Configuration management for SmolLM3 fine-tuning
"""
import os
import sys
import importlib.util
from typing import Any
# Add the project root to Python path
project_root = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
if project_root not in sys.path:
sys.path.insert(0, project_root)
# Add config directory to path
config_dir = os.path.join(project_root, 'config')
if config_dir not in sys.path:
sys.path.insert(0, config_dir)
try:
from config.train_smollm3 import SmolLM3Config, get_config as get_default_config
except ImportError:
# Fallback: try direct import
import sys
sys.path.insert(0, os.path.join(project_root, 'config'))
from train_smollm3 import SmolLM3Config, get_config as get_default_config
def get_config(config_path: str) -> SmolLM3Config:
"""Load configuration from file or return default"""
if os.path.exists(config_path):
# Load from file if it exists
spec = importlib.util.spec_from_file_location("config_module", config_path)
config_module = importlib.util.module_from_spec(spec)
spec.loader.exec_module(config_module)
if hasattr(config_module, 'config'):
return config_module.config
else:
# Try to find a config class
for attr_name in dir(config_module):
attr = getattr(config_module, attr_name)
if isinstance(attr, SmolLM3Config):
return attr
# Return default configuration
return get_default_config(config_path) |