From 6ff192651684f351d8906a5f3e17e4451ca21133 Mon Sep 17 00:00:00 2001 From: SkalskiP Date: Tue, 3 Sep 2024 13:00:18 +0200 Subject: [PATCH] fix: current Florence-2 training pipeline is missing `timm` --- requirements/requirements.txt | 1 + 1 file changed, 1 insertion(+) diff --git a/requirements/requirements.txt b/requirements/requirements.txt index ce70d29..c10deb3 100644 --- a/requirements/requirements.txt +++ b/requirements/requirements.txt @@ -7,3 +7,4 @@ sentencepiece~=0.2.0 peft~=0.12.0 flash-attn~=2.6.3 einops~=0.8.0 +timm~=1.0.9 \ No newline at end of file