13 lines
191 B
YAML
13 lines
191 B
YAML
|
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
|