From 14de324c5807bd6efb4ff7b4f47a04fdf6cf49f2 Mon Sep 17 00:00:00 2001
From: clemsgrs <clementnet@outlook.com>
Date: Wed, 25 Feb 2026 16:51:25 +0000
Subject: [PATCH] minor fixes

---
 slide2vec/models/models.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/slide2vec/models/models.py b/slide2vec/models/models.py
index 0f835c8..cc93daf 100644
--- a/slide2vec/models/models.py
+++ b/slide2vec/models/models.py
@@ -711,7 +711,7 @@ def forward(self, x):
 
 class Midnight12k(FeatureExtractor):
     def __init__(self):
-        self.features_dim = 1536
+        self.features_dim = 3072
         super(Midnight12k, self).__init__()
 
     def build_encoder(self):
@@ -749,7 +749,7 @@ def build_encoder(self):
 
     def get_transforms(self):
         return AutoImageProcessor.from_pretrained(
-            "histai/hibou-L", trust_remote_code=True
+            f"histai/{self.arch}", trust_remote_code=True
         )
 
     def forward(self, x):