name: mini-transformer
channels:
  - defaults
dependencies:
  - python=3.11
  - pip
  - pip:
      - --extra-index-url https://download.pytorch.org/whl/cu121
      - torch==2.3.1
      - numpy
      - transformers
      - omegaconf
      - hydra-core
      - tokenizers
      - datasets
      - huggingface_hub
      - matplotlib
      - jupyterlab
      - fastapi
      - uvicorn[standard]
      - chainlit
      - pytest
      - pytest-cov
      - ruff
      - black
      - mypy
      - pre-commit
      - httpx
  - evaluate
  - sacrebleu
  - rouge-score
  - bert-score
  - tqdm