gpt-oss-20b-mlflow / conda.yaml
MLProductionization's picture
Update conda.yaml
f5ea19f verified
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