channels: | |
- conda-forge | |
dependencies: | |
- python=3.12.3 | |
- pip<=25.2 | |
- pip: | |
- mlflow==3.2.0 | |
- accelerate==1.10.0 | |
- torch==2.8.0 | |
- torchvision==0.23.0 | |
- transformers==4.55.0 | |
name: mlflow-env | |
channels: | |
- conda-forge | |
dependencies: | |
- python=3.12.3 | |
- pip<=25.2 | |
- pip: | |
- mlflow==3.2.0 | |
- accelerate==1.10.0 | |
- torch==2.8.0 | |
- torchvision==0.23.0 | |
- transformers==4.55.0 | |
name: mlflow-env | |