KoboldAI-Client/environments/rocm.yml

name: koboldai
channels:
  - huggingface
  - conda-forge
  - defaults
dependencies:
  - transformers
  - colorama
  - flask-socketio
  - python=3.8.*
  - pip
  - git
  - pip:
    - --find-links https://download.pytorch.org/whl/rocm4.2/torch_stable.html
    - torch
    - torchvision==0.11.1
    - flask-cloudflared
    - flask-ngrok
ROCm Conda Allows anyone to easily create a ROCm compatible conda environment. Currently set to the newer transformers, you can edit the github link if you want the finetune one. 2021-11-17 22:15:01 +01:00			`name: koboldai`
			`channels:`
Migrate to official transformers No longer using VE's fork since its obsolete, in line with what we already did for the CUDA varient. 2021-12-07 23:35:28 +01:00			`- huggingface`
ROCm Conda Allows anyone to easily create a ROCm compatible conda environment. Currently set to the newer transformers, you can edit the github link if you want the finetune one. 2021-11-17 22:15:01 +01:00			`- conda-forge`
			`- defaults`
			`dependencies:`
Migrate to official transformers No longer using VE's fork since its obsolete, in line with what we already did for the CUDA varient. 2021-12-07 23:35:28 +01:00			`- transformers`
ROCm Conda Allows anyone to easily create a ROCm compatible conda environment. Currently set to the newer transformers, you can edit the github link if you want the finetune one. 2021-11-17 22:15:01 +01:00			`- colorama`
			`- flask-socketio`
			`- python=3.8.*`
			`- pip`
			`- git`
			`- pip:`
			`- --find-links https://download.pytorch.org/whl/rocm4.2/torch_stable.html`
			`- torch`
			`- torchvision==0.11.1`
			`- flask-cloudflared`
Migrate to official transformers No longer using VE's fork since its obsolete, in line with what we already did for the CUDA varient. 2021-12-07 23:35:28 +01:00			`- flask-ngrok`