From 9f3e406d3a61a107bbeaf28722e524a8407e4843 Mon Sep 17 00:00:00 2001 From: Henk Date: Mon, 11 Sep 2023 13:38:18 +0200 Subject: [PATCH] xformers in cuda env --- environments/huggingface.yml | 2 ++ 1 file changed, 2 insertions(+) diff --git a/environments/huggingface.yml b/environments/huggingface.yml index a4936bf7..cf4722bf 100644 --- a/environments/huggingface.yml +++ b/environments/huggingface.yml @@ -2,6 +2,7 @@ name: koboldai channels: - pytorch - nvidia + - xformers - conda-forge - defaults dependencies: @@ -26,6 +27,7 @@ dependencies: - termcolor - Pillow - psutil + - xformers=0.0.21 - pip: - flask-cloudflared==0.0.10 - flask-ngrok