Remove an unnecessary lib

brain-score · Jan 14, 2025 · 63e0f4b · 63e0f4b
1 parent 551c577
commit 63e0f4b
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 6 deletions.
diff --git a/brainscore_vision/models/timm_models/model.py b/brainscore_vision/models/timm_models/model.py
@@ -4,16 +4,12 @@
 from pathlib import Path
 import ssl
 
-import torchvision.models
-import torch
-
-from brainscore_vision.model_helpers.activations.pytorch import PytorchWrapper
-
 import timm
 import numpy as np
 import torchvision.transforms as T
 from PIL import Image
 
+from brainscore_vision.model_helpers.activations.pytorch import PytorchWrapper
 
 # Disable SSL verification 
 ssl._create_default_https_context = ssl._create_unverified_context

diff --git a/brainscore_vision/models/timm_models/requirements.txt b/brainscore_vision/models/timm_models/requirements.txt
@@ -1,4 +1,3 @@
 torch
 torchvision
-albumentations
 timm