environment.yml

# 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