From 354c01d65c4fceeaf743e6e8c01301245f69a74d Mon Sep 17 00:00:00 2001 From: Bingxin Ke Date: Sun, 28 Apr 2024 13:52:23 +0200 Subject: [PATCH] [FIX] torchvision requirement #59 --- requirements+.txt | 3 +-- requirements.txt | 1 + 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/requirements+.txt b/requirements+.txt index d914638..ed03f66 100644 --- a/requirements+.txt +++ b/requirements+.txt @@ -1,4 +1,3 @@ omegaconf pandas -tabulate -torchvision==0.15.2 \ No newline at end of file +tabulate \ No newline at end of file diff --git a/requirements.txt b/requirements.txt index fff7af6..33cc65a 100644 --- a/requirements.txt +++ b/requirements.txt @@ -3,4 +3,5 @@ diffusers>=0.25.0 matplotlib scipy torch==2.0.1 +torchvision==0.15.2 transformers>=4.32.1 \ No newline at end of file