rocm.yml

name: koboldai
channels:
  - conda-forge
  - defaults
dependencies:
  - colorama
  - flask=2.2.3
  - flask-socketio=5.3.2
  - flask-session=0.4.0
  - python-socketio=5.7.2
  - python=3.8.*
  - eventlet=0.33.3
  - dnspython=2.2.1
  - markdown
  - bleach=4.1.0
  - pip
  - git=2.35.1
  - sentencepiece
  - protobuf
  - marshmallow>=3.13
  - apispec-webframeworks
  - loguru
  - termcolor
  - psutil
  - pip:
    - --extra-index-url https://download.pytorch.org/whl/rocm5.1.1
    - torch==1.12.1+rocm5.1.1
    - flask-cloudflared==0.0.10
    - flask-ngrok
    - Werkzeug==2.3.7
    - lupa==1.10
    - transformers==4.24.0
    - huggingface_hub==0.12.1
    - safetensors
    - accelerate
    - git+https://github.com/VE-FORBRYDERNE/mkultra