From 91155ed2f303704b9cd77ff71a5aa60339bf832e Mon Sep 17 00:00:00 2001 From: Henk Date: Wed, 23 Aug 2023 20:34:40 +0200 Subject: [PATCH] HF dependencies --- environments/huggingface.yml | 3 ++- environments/rocm.yml | 3 ++- requirements.txt | 3 ++- 3 files changed, 6 insertions(+), 3 deletions(-) diff --git a/environments/huggingface.yml b/environments/huggingface.yml index db6b931c..1e54011b 100644 --- a/environments/huggingface.yml +++ b/environments/huggingface.yml @@ -34,7 +34,8 @@ dependencies: - lupa==1.10 - transformers==4.32.* - huggingface_hub==0.16.4 - - git+https://github.com/huggingface/optimum@2c1eaf6c0242badb86f3d626811ee3d7d220ec06 + - optimum==1.12.0 + - safetensors==0.3.2 - accelerate==0.21.0 - git+https://github.com/VE-FORBRYDERNE/mkultra - flask-session diff --git a/environments/rocm.yml b/environments/rocm.yml index bf494799..22aff85c 100644 --- a/environments/rocm.yml +++ b/environments/rocm.yml @@ -32,7 +32,8 @@ dependencies: - lupa==1.10 - transformers==4.32.* - huggingface_hub==0.16.4 - - git+https://github.com/huggingface/optimum@2c1eaf6c0242badb86f3d626811ee3d7d220ec06 + - optimum==1.12.0 + - safetensors==0.3.2 - accelerate==0.21.0 - git+https://github.com/VE-FORBRYDERNE/mkultra - ansi2html diff --git a/requirements.txt b/requirements.txt index b3ba1d3f..4f0494c8 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,6 +1,7 @@ transformers==4.32.* huggingface_hub==0.16.4 -git+https://github.com/huggingface/optimum@2c1eaf6c0242badb86f3d626811ee3d7d220ec06 +optimum==1.12.0 +safetensors==0.3.2 Flask==2.2.3 Flask-SocketIO==5.3.2 python-socketio==5.7.2