From 3a219e25021ff7ec46830c3bc0597c5183b25293 Mon Sep 17 00:00:00 2001 From: Victor Hall Date: Fri, 22 Mar 2024 12:41:14 -0400 Subject: [PATCH] update docker to fix cog --- docker/requirements-build.txt | 2 +- docker/requirements-runtime.txt | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/docker/requirements-build.txt b/docker/requirements-build.txt index 8670b40..46132f2 100644 --- a/docker/requirements-build.txt +++ b/docker/requirements-build.txt @@ -5,4 +5,4 @@ omegaconf==2.2.3 protobuf==3.20.3 pyre-extensions==0.0.29 pytorch-lightning==1.9.2 -transformers==4.35.0 +transformers==4.38.2 diff --git a/docker/requirements-runtime.txt b/docker/requirements-runtime.txt index 23a502a..a9c8761 100644 --- a/docker/requirements-runtime.txt +++ b/docker/requirements-runtime.txt @@ -1,5 +1,5 @@ aiohttp==3.8.4 -bitsandbytes==0.41.1 +bitsandbytes==0.43.0 scipy colorama==0.4.6 compel~=1.1.3