diff --git a/environments/huggingface.yml b/environments/huggingface.yml index a5affd40..9fd2ce5f 100644 --- a/environments/huggingface.yml +++ b/environments/huggingface.yml @@ -30,6 +30,7 @@ dependencies: - lupa==1.10 - transformers==4.24.0 - huggingface_hub>=0.10.1 + - safetensors - accelerate - git+https://github.com/VE-FORBRYDERNE/mkultra - flask-session diff --git a/environments/rocm.yml b/environments/rocm.yml index 2318b123..ad39a3a6 100644 --- a/environments/rocm.yml +++ b/environments/rocm.yml @@ -30,6 +30,7 @@ dependencies: - lupa==1.10 - transformers==4.24.0 - huggingface_hub>=0.10.1 + - safetensors - accelerate - git+https://github.com/VE-FORBRYDERNE/mkultra - ansi2html diff --git a/install_requirements.sh b/install_requirements.sh index 7d2c0c2f..6f0e0dfd 100755 --- a/install_requirements.sh +++ b/install_requirements.sh @@ -1,4 +1,5 @@ #!/bin/bash +git submodule update --init --recursive if [[ $1 = "cuda" ]]; then wget -qO- https://micromamba.snakepit.net/api/micromamba/linux-64/latest | tar -xvj bin/micromamba bin/micromamba create -f environments/huggingface.yml -r runtime -n koboldai -y diff --git a/requirements.txt b/requirements.txt index c1429449..53e450c9 100644 --- a/requirements.txt +++ b/requirements.txt @@ -27,4 +27,5 @@ flask_compress ijson bitsandbytes ftfy -pydub \ No newline at end of file +pydub +safetensors \ No newline at end of file diff --git a/requirements_mtj.txt b/requirements_mtj.txt index 2a103144..39c8451e 100644 --- a/requirements_mtj.txt +++ b/requirements_mtj.txt @@ -20,6 +20,7 @@ bleach==4.1.0 flask-session marshmallow>=3.13 apispec-webframeworks +safetensors loguru Pillow diffusers