#!/bin/bash
# 安装三个时间序列基础模型

echo "=================================="
echo "安装时间序列基础模型"
echo "=================================="

# 激活conda环境
source /Users/xxz/miniforge3/bin/activate
conda activate crypto_pred_312

echo ""
echo "[1/3] 安装 Google TimesFM 2.5..."
pip install git+https://github.com/google-research/timesfm.git

echo ""
echo "[2/3] 安装 IBM TTM-R2 (Granite TSFM)..."
pip install tsfm_public

echo ""
echo "[3/3] 安装 Lag-Llama 依赖..."
pip install gluonts[torch] torch

echo ""
echo "=================================="
echo "✓ 所有模型安装完成!"
echo "=================================="
echo ""
echo "模型位置："
echo "  - TimesFM 2.5: HuggingFace自动下载"
echo "  - TTM-R2: ibm-granite/granite-timeseries-ttm-r2"
echo "  - Lag-Llama: time-series-foundation-models/lag-llama"