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 | |
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 | |