gpt-local / environment.yml
DRDELATV's picture
Upload folder using huggingface_hub
22ca508 verified
# Conda environment file for GPT Local
name: gpt-local
channels:
- pytorch
- conda-forge
- defaults
dependencies:
- python=3.9
- pytorch>=2.0.0
- pytorch-cpu # or pytorch-cuda for GPU
- pip
- pip:
- transformers>=4.21.0
- gradio>=3.50.0
- accelerate>=0.20.0
- sentencepiece>=0.1.99
- numpy>=1.21.0
- pandas>=1.3.0
- matplotlib>=3.5.0
- requests>=2.25.0
- tqdm>=4.60.0