From 5db8156747f82d67ea3f70e771619c53ae85d795 Mon Sep 17 00:00:00 2001 From: Nicola Dall'Asen Date: Wed, 13 Mar 2024 12:30:50 +0100 Subject: [PATCH] use transformers 4.38.1 instead of 4.38.2 as autocast fails on mps on latest version --- pyproject.toml | 2 +- requirements.txt | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/pyproject.toml b/pyproject.toml index a391008..da88b54 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -13,7 +13,7 @@ readme = "README.md" requires-python = ">=3.8" dependencies = [ "torch>=2.0.1", - "transformers>=4.38.2", + "transformers==4.38.1", "timm>=0.9.16", "gradio>=4.13.0", "accelerate", diff --git a/requirements.txt b/requirements.txt index 7a93ed8..7535a74 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,5 +1,5 @@ torch>=2.0.1 -transformers>=4.38.2 +transformers==4.38.1 timm>=0.9.16 gradio>=4.13.0 accelerate