From 4699ded3ce4bc23992cb3c8f0ec02c25594d2c37 Mon Sep 17 00:00:00 2001 From: Henk Date: Sat, 22 Oct 2022 19:00:06 +0200 Subject: [PATCH] Tuner Dependencies --- environments/huggingface.yml | 2 ++ environments/rocm.yml | 2 ++ requirements.txt | 2 ++ 3 files changed, 6 insertions(+) diff --git a/environments/huggingface.yml b/environments/huggingface.yml index 7abceefa..9957f1c5 100644 --- a/environments/huggingface.yml +++ b/environments/huggingface.yml @@ -20,9 +20,11 @@ dependencies: - marshmallow>=3.13 - apispec-webframeworks - loguru + - termcolor - pip: - flask-cloudflared - flask-ngrok - lupa==1.10 - transformers>=4.20.1 - accelerate + - git+https://github.com/VE-FORBRYDERNE/mkultra diff --git a/environments/rocm.yml b/environments/rocm.yml index e885f4df..c6fea0b3 100644 --- a/environments/rocm.yml +++ b/environments/rocm.yml @@ -17,6 +17,7 @@ dependencies: - marshmallow>=3.13 - apispec-webframeworks - loguru + - termcolor - pip: - --extra-index-url https://download.pytorch.org/whl/rocm5.1.1 - torch @@ -26,3 +27,4 @@ dependencies: - lupa==1.10 - transformers>=4.20.1 - accelerate + - git+https://github.com/VE-FORBRYDERNE/mkultra diff --git a/requirements.txt b/requirements.txt index b1e2247c..0d2bcd6b 100644 --- a/requirements.txt +++ b/requirements.txt @@ -16,3 +16,5 @@ flask-session marshmallow>=3.13 apispec-webframeworks loguru +termcolor +git+https://github.com/VE-FORBRYDERNE/mkultra \ No newline at end of file