LLaVA/llava.yml

channels:
  - defaults      # 使用默认的 channels（如有特殊需求，可加入其他 channels）
dependencies:
  - python=3.11   # Python 版本

  # 安装通过 pip 安装的 Python 库
  - pip
  - pip:
    - torch==2.0.1
    - accelerate==0.21.0
    - bitsandbytes==0.41.0
    - deepspeed==0.9.5
    - einops-exts==0.0.4
    - einops==0.6.1
    - gradio==3.35.2
    - gradio_client==0.2.9
    - httpx==0.24.0
    - markdown2==2.4.10
    - numpy==1.26.0
    - peft==0.4.0
    - scikit-learn==1.2.2
    - sentencepiece==0.1.99
    - shortuuid==1.0.11
    - timm==0.6.13
    - tokenizers==0.13.3
    - torchvision==0.15.2
    - transformers==4.31.0
    - wandb==0.15.12
    - wavedrom==2.0.3.post3
    - Pygments==2.16.1
    - opencv-python
    - pycocotools

# 如果需要添加运行命令或其他脚本执行步骤，可以在 run 中定义