From f4f996681ca340e940315ca0ebc162c655904a7d Mon Sep 17 00:00:00 2001 From: Volpeon Date: Wed, 5 Apr 2023 16:02:04 +0200 Subject: Add color jitter --- environment.yaml | 1 + 1 file changed, 1 insertion(+) (limited to 'environment.yaml') diff --git a/environment.yaml b/environment.yaml index 1de76bd..418cb22 100644 --- a/environment.yaml +++ b/environment.yaml @@ -24,5 +24,6 @@ dependencies: - safetensors==0.3.0 - setuptools==65.6.3 - test-tube>=0.7.5 + - timm==0.8.17.dev0 - transformers==4.27.1 - triton==2.0.0.post1 -- cgit v1.2.3-54-g00ecf