env.yaml

name: ryzenai-transformers
channels:
  - conda-forge
  - defaults
dependencies:
  - antlr
  - benchmark
  - boost
  - boost-cpp
  - clang
  - clang-tools
  - clangdev
  - clangxx
  - cmake
  - cxxopts
  - eigen
  - ffmpeg
  - gtest
  - graphviz
  - llvm-tools
  - llvmdev
  - m2w64-gcc-libgfortran
  - m2w64-gcc-libs
  - m2w64-gcc-libs-core
  - m2w64-gmp
  - m2w64-libwinpthread-git
  - msys2-conda-epoch
  - nlohmann_json
  - numpy=1.26.4
  - pip
  - pre-commit
  - pytest
  - python=3.11
  - scipy
  - setuptools
  - simplejson
  - spdlog
  - symlink-exe-runtime
  - vc
  - pip:
    - onnx
    - onnxruntime
    - torch==2.1.2
    - torchmetrics
    - torchvision==0.16.2
    - tqdm
    - huggingface-hub==0.21.4
    - transformers==4.37.2
    - optimum==1.18.0
    - tokenizers
    - diffusers
    - datasets
    - accelerate
    - pybind11[global]
    - nanobind
    - matplotlib
    - autopep8
    - bokeh
    - clang-format
    - opencv-python
    - scikit-learn
    - thop
    - tabulate
    - black
    - tiktoken
    - einops
    - human-eval