deploy.py

# FantasyTalking部署脚本

import os
import subprocess
import sys
from pathlib import Path


def check_gpu():
    """检查GPU可用性"""
    try:
        import torch
        if torch.cuda.is_available():
            gpu_count = torch.cuda.device_count()
            gpu_name = torch.cuda.get_device_name(0) if gpu_count > 0 else "Unknown"
            gpu_memory = torch.cuda.get_device_properties(0).total_memory // (1024**3) if gpu_count > 0 else 0
            
            print(f"✅ GPU可用: {gpu_name}")
            print(f"✅ GPU内存: {gpu_memory}GB")
            
            if gpu_memory < 5:
                print("⚠️  警告: GPU内存可能不足，建议至少5GB VRAM")
            
            return True
        else:
            print("❌ 未检测到可用的GPU")
            return False
    except ImportError:
        print("❌ PyTorch未安装")
        return False


def install_dependencies():
    """安装依赖"""
    print("📦 安装依赖包...")
    subprocess.check_call([sys.executable, "-m", "pip", "install", "-r", "requirements.txt"])
    print("✅ 依赖安装完成")


def download_models():
    """下载模型（需要huggingface-cli）"""
    print("📥 开始下载模型...")
    
    models_dir = Path("./models")
    models_dir.mkdir(exist_ok=True)
    
    # 检查huggingface-cli
    try:
        subprocess.check_call(["huggingface-cli", "--help"], stdout=subprocess.DEVNULL)
    except (subprocess.CalledProcessError, FileNotFoundError):
        print("安装huggingface-hub[cli]...")
        subprocess.check_call([sys.executable, "-m", "pip", "install", "huggingface_hub[cli]"])
    
    # 下载模型
    models_to_download = [
        ("Wan-AI/Wan2.1-I2V-14B-720P", "./models/Wan2.1-I2V-14B-720P"),
        ("facebook/wav2vec2-base-960h", "./models/wav2vec2-base-960h"),
    ]
    
    for model_id, local_dir in models_to_download:
        print(f"下载 {model_id}...")
        subprocess.check_call([
            "huggingface-cli", "download", model_id, 
            "--local-dir", local_dir
        ])
    
    # 下载FantasyTalking权重
    print("下载FantasyTalking权重...")
    subprocess.check_call([
        "huggingface-cli", "download", "acvlab/FantasyTalking", 
        "fantasytalking_model.ckpt", "--local-dir", "./models"
    ])
    
    print("✅ 模型下载完成")


def check_model_files():
    """检查模型文件"""
    required_files = [
        "./models/Wan2.1-I2V-14B-720P",
        "./models/wav2vec2-base-960h", 
        "./models/fantasytalking_model.ckpt"
    ]
    
    missing_files = []
    for file_path in required_files:
        if not os.path.exists(file_path):
            missing_files.append(file_path)
    
    if missing_files:
        print("❌ 缺少以下模型文件:")
        for file in missing_files:
            print(f"   - {file}")
        return False
    else:
        print("✅ 所有模型文件已就绪")
        return True


def start_app():
    """启动应用"""
    print("🚀 启动FantasyTalking应用...")
    subprocess.check_call([sys.executable, "app.py"])


def main():
    """主函数"""
    print("🎬 FantasyTalking 自动部署脚本")
    print("=" * 50)
    
    # 检查GPU
    if not check_gpu():
        print("⚠️  继续使用CPU模式（速度会很慢）")
    
    # 安装依赖
    install_dependencies()
    
    # 检查模型文件
    if not check_model_files():
        print("📥 需要下载模型文件...")
        download_models()
    
    print("✅ 部署完成！")
    print("\n启动应用...")
    start_app()


if __name__ == "__main__":
    main()