diff --git a/requirements.txt b/requirements.txt index 8818cc5..35cbf26 100644 --- a/requirements.txt +++ b/requirements.txt @@ -4,7 +4,7 @@ gradio_client==0.2.5 gradio==3.33.1 pydantic==1.10.12 -accelerate==0.22.* +accelerate==0.23.* colorama datasets einops diff --git a/requirements_nocuda.txt b/requirements_nocuda.txt index de02295..d4eb041 100644 --- a/requirements_nocuda.txt +++ b/requirements_nocuda.txt @@ -4,7 +4,7 @@ gradio_client==0.2.5 gradio==3.33.1 pydantic==1.10.12 -accelerate==0.22.* +accelerate==0.23.* colorama datasets einops