From 6c22f723e22d4e335a23e41ab58c9a874a8b1d01 Mon Sep 17 00:00:00 2001 From: bmaltais Date: Sat, 3 Dec 2022 18:31:12 -0500 Subject: [PATCH] Update requirements.txt --- requirements.txt | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/requirements.txt b/requirements.txt index 4c69950..3839640 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,4 +1,4 @@ -accelerate +accelerate==0.14.0 transformers>=4.21.0 ftfy albumentations @@ -7,4 +7,4 @@ einops diffusers[torch]==0.9.0 pytorch_lightning bitsandbytes==0.35.0 -tensorboard \ No newline at end of file +tensorboard