diff --git a/environments/rocm.yml b/environments/rocm.yml index bbcb810b..782d9903 100644 --- a/environments/rocm.yml +++ b/environments/rocm.yml @@ -1,8 +1,10 @@ name: koboldai channels: + - huggingface - conda-forge - defaults dependencies: + - transformers - colorama - flask-socketio - python=3.8.* @@ -13,5 +15,4 @@ dependencies: - torch - torchvision==0.11.1 - flask-cloudflared - - git+https://github.com/vfbd/transformers#tfs - - flask-ngrok \ No newline at end of file + - flask-ngrok diff --git a/play.sh b/play.sh new file mode 100644 index 00000000..bd87baa1 --- /dev/null +++ b/play.sh @@ -0,0 +1,4 @@ +#!/bin/bash +conda env create -f environments/huggingface.yml +conda activate koboldai +python aiserver.py $* \ No newline at end of file