Upload config/config_loader.py with huggingface_hub

config/config_loader.py

@@ -0,0 +1,142 @@
+import os
+import yaml
+import logging
+from typing import Dict, Any, Optional
+from pydantic_settings import BaseSettings
+from dotenv import load_dotenv
+
+logger = logging.getLogger(__name__)
+
+# 加载环境变量
+load_dotenv()
+
+class AppSettings(BaseSettings):
+    """
+    应用配置类
+    """
+    # 模型配置
+    model_name: str = os.getenv("MODEL_NAME", "qwen/qwen-image-2512")
+    model_revision: str = os.getenv("MODEL_REVISION", "main")
+    model_dir: str = os.getenv("MODEL_DIR", "./models")
+    cache_dir: Optional[str] = os.getenv("CACHE_DIR")
+    
+    # 推理配置
+    device: str = os.getenv("DEVICE", "cpu")
+    max_memory_cpu: str = os.getenv("MAX_MEMORY_CPU", "16GB")
+    num_threads: int = int(os.getenv("NUM_THREADS", "4"))
+    num_interop_threads: int = int(os.getenv("NUM_INTEROP_THREADS", "2"))
+    
+    # API配置
+    request_limit: int = int(os.getenv("REQUEST_LIMIT", "100"))
+    request_window: int = int(os.getenv("REQUEST_WINDOW", "60"))
+    
+    # 日志配置
+    log_level: str = os.getenv("LOG_LEVEL", "info")
+    log_format: str = os.getenv("LOG_FORMAT", "%(asctime)s - %(name)s - %(levelname)s - %(message)s")
+    
+    class Config:
+        case_sensitive = False
+        env_file = ".env"
+
+def load_config(config_path: str = "config.yaml") -> Dict[str, Any]:
+    """
+    加载YAML配置文件
+    
+    Args:
+        config_path: 配置文件路径
+        
+    Returns:
+        Dict[str, Any]: 配置字典
+    """
+    if not os.path.exists(config_path):
+        logger.warning(f"配置文件不存在: {config_path}，使用默认配置")
+        return {}
+    
+    try:
+        with open(config_path, "r", encoding="utf-8") as f:
+            config = yaml.safe_load(f)
+        
+        logger.info(f"配置文件加载成功: {config_path}")
+        return config
+        
+    except Exception as e:
+        logger.error(f"配置文件加载失败: {str(e)}，使用默认配置")
+        return {}
+
+def get_settings() -> AppSettings:
+    """
+    获取应用设置
+    
+    Returns:
+        AppSettings: 应用设置实例
+    """
+    return AppSettings()
+
+def merge_configs(default_config: Dict[str, Any], user_config: Dict[str, Any]) -> Dict[str, Any]:
+    """
+    合并配置
+    
+    Args:
+        default_config: 默认配置
+        user_config: 用户配置
+        
+    Returns:
+        Dict[str, Any]: 合并后的配置
+    """
+    merged = default_config.copy()
+    
+    for key, value in user_config.items():
+        if key in merged and isinstance(merged[key], dict) and isinstance(value, dict):
+            merged[key] = merge_configs(merged[key], value)
+        else:
+            merged[key] = value
+    
+    return merged
+
+# 默认配置
+DEFAULT_CONFIG = {
+    "model": {
+        "name": "qwen/qwen-image-2512",
+        "revision": "main",
+        "dir": "./models",
+        "cache_dir": "~/.cache/huggingface/hub",
+        "verify_integrity": True
+    },
+    "inference": {
+        "device": "cpu",
+        "dtype": "float32",
+        "max_memory": {
+            "cpu": "16GB"
+        },
+        "num_threads": 4,
+        "num_interop_threads": 2,
+        "enable_attention_slicing": True,
+        "enable_sequential_cpu_offload": True,
+        "low_cpu_mem_usage": True
+    },
+    "api": {
+        "host": "0.0.0.0",
+        "port": 7860,
+        "workers": 1,
+        "docs_url": "/docs",
+        "redoc_url": "/redoc",
+        "request_limit": 100,
+        "request_window": 60,
+        "cors_origins": ["*"]
+    },
+    "ui": {
+        "title": "Qwen-Image-2512 文本到图像生成",
+        "theme": "soft",
+        "max_images": 4
+    },
+    "logging": {
+        "level": "info",
+        "format": "%(asctime)s - %(name)s - %(levelname)s - %(message)s"
+    }
+}
+
+# 加载配置
+CONFIG = merge_configs(DEFAULT_CONFIG, load_config())
+
+# 获取设置实例
+SETTINGS = get_settings()