channels:
- conda-forge
dependencies:
- python=3.11.13
- pip
- pip:
  - mlflow==3.1.4
  - accelerate==1.9.0
  - torch==2.7.1
  - torchvision==0.22.1
  - transformers==4.54.0
name: mlflow-env