MLProductionization's picture
Initial model upload
6226552 verified
channels:
- conda-forge
dependencies:
- python=3.12.3
- pip<=24.0
- pip:
- mlflow==3.4.0
- accelerate==1.5.2
- torch==2.6.0
- torchvision==0.21.0
- transformers==4.50.2
name: mlflow-env