fixes model_helpers import

brain-score · May 16, 2024 · b5689c5 · b5689c5
1 parent 12f1e1c
commit b5689c5
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/brainscore_vision/models/cvt_cvt-w24-384-in22k_finetuned-in1k_4/model.py b/brainscore_vision/models/cvt_cvt-w24-384-in22k_finetuned-in1k_4/model.py
@@ -1,7 +1,7 @@
-from model_helpers.check_submission import check_models
+from brainscore_vision.model_helpers.check_submission import check_models
 import functools
 from transformers import AutoFeatureExtractor, CvtForImageClassification
-from model_helpers.activations.pytorch import PytorchWrapper
+from brainscore_vision.model_helpers.activations.pytorch import PytorchWrapper
 from PIL import Image
 import numpy as np
 import torch