theliu
97 lines
3.2 KiB
Python
97 lines
3.2 KiB
Python
"""
|
|
config.py - All configuration for VidMarmot.
|
|
|
|
API keys, model paths, default parameters — everything lives here.
|
|
Edit this file directly to configure your setup.
|
|
"""
|
|
|
|
import os
|
|
|
|
BASE_DIR = os.path.dirname(os.path.abspath(__file__))
|
|
|
|
|
|
# ========== ASR Model ==========
|
|
# Default: project_dir/models/qwen/Qwen3-ForcedAligner-0.6B
|
|
# Override via env var VIDMARMOT_ASR_MODEL_DIR
|
|
ASR_MODEL_DIR = os.path.join(BASE_DIR, "models", "qwen", "Qwen3-ForcedAligner-0.6B").replace("\\", "/")
|
|
|
|
|
|
# ========== LLM Providers (scene planning / text generation) ==========
|
|
# Fill in your API keys below. Only providers with keys will be usable.
|
|
LLM_PROVIDERS = {
|
|
"Qwen3.5-35B (ModelScope)": {
|
|
"api_key": "", # ← put your ModelScope API key here
|
|
"api_base": "https://api-inference.modelscope.cn/v1/",
|
|
"model": "Qwen/Qwen3.5-35B-A3B",
|
|
},
|
|
"GLM-4-9B (SiliconFlow)": {
|
|
"api_key": "", # ← put your SiliconFlow API key here
|
|
"api_base": "https://api.siliconflow.cn/v1/",
|
|
"model": "THUDM/glm-4-9b-chat",
|
|
},
|
|
"Qwen3-32B (SiliconFlow)": {
|
|
"api_key": "", # ← put your SiliconFlow API key here
|
|
"api_base": "https://api.siliconflow.cn/v1/",
|
|
"model": "Qwen/Qwen3-32B",
|
|
},
|
|
"GLM-5 (ModelScope)": {
|
|
"api_key": "", # ← put your ModelScope API key here
|
|
"api_base": "https://api-inference.modelscope.cn/v1/",
|
|
"model": "ZhipuAI/GLM-5",
|
|
},
|
|
"Qwen3-235B-A22B (Aliyun)": {
|
|
"api_key": "", # ← put your Aliyun DashScope API key here
|
|
"api_base": "https://dashscope.aliyuncs.com/compatible-mode/v1/",
|
|
"model": "qwen3-235b-a22b",
|
|
},
|
|
"DeepSeek-V3": {
|
|
"api_key": "", # ← put your DeepSeek API key here
|
|
"api_base": "https://api.deepseek.com/v1/",
|
|
"model": "deepseek-chat",
|
|
},
|
|
"DeepSeek-R1": {
|
|
"api_key": "", # ← put your DeepSeek API key here
|
|
"api_base": "https://api.deepseek.com/v1/",
|
|
"model": "deepseek-reasoner",
|
|
},
|
|
"OpenAI (Custom Router)": {
|
|
"api_key": "", # ← put your OpenAI-compatible API key here
|
|
"api_base": "https://api.openai.com/v1/", # change if using a custom router
|
|
"model": "gpt-4o",
|
|
},
|
|
}
|
|
|
|
|
|
# ========== Text-to-Image Models ==========
|
|
# Fill in your API keys below.
|
|
IMAGE_MODELS = {
|
|
"Kolors (SiliconFlow)": {
|
|
"provider": "siliconflow",
|
|
"api_key": "", # ← put your SiliconFlow API key here
|
|
"api_base": "https://api.siliconflow.cn/v1/images/generations",
|
|
"model": "Kwai-Kolors/Kolors",
|
|
"default_size": "1280x720",
|
|
"guidance_scale": 7.5,
|
|
},
|
|
"Qwen-Image (ModelScope)": {
|
|
"provider": "modelscope",
|
|
"api_key": "", # ← put your ModelScope API key here
|
|
"api_base": "https://api-inference.modelscope.cn/v1/images/generations",
|
|
"poll_interval": 3,
|
|
"max_wait": 180,
|
|
"model": "Qwen/Qwen-Image-2512",
|
|
"default_size": "1280x720",
|
|
"guidance_scale": 7.5,
|
|
},
|
|
}
|
|
|
|
DEFAULT_IMAGE_MODEL = "Kolors (SiliconFlow)"
|
|
|
|
|
|
# ========== Defaults ==========
|
|
DEFAULT_FPS = 24
|
|
DEFAULT_VIDEO_SIZE = "1280x720"
|
|
|
|
# Negative prompt for image generation
|
|
NEGATIVE_PROMPT = "blurry, low quality, deformed, text, letters, words, subtitle, logo, watermark, caption, label, number"
|