OLive/examples/llama2/conda_gpu.yaml

23 строки
479 B
YAML

name: project_environment
channels:
- defaults
dependencies:
- python=3.8.13
- pip=22.3.1
- pip:
- accelerate
- bitsandbytes
- peft
- scikit-learn
- sentencepiece
- datasets
- evaluate
- psutil
- optimum
- scipy
- scikit-learn
- onnxruntime-genai
- torch
- --extra-index-url https://download.pytorch.org/whl/cu118
- git+https://github.com/microsoft/Olive#egg=olive-ai[gpu,azureml]