From 026c1c653dfde4d91fe199bfda28d7de1d680ed0 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?L=C3=BD=20Tu=E1=BA=A5n=20L=E1=BB=99c?= <119644050+ComTamSuonBiCha@users.noreply.github.com> Date: Tue, 7 Apr 2026 14:24:00 +0700 Subject: [PATCH] Fix layer module reference in image feature extractor MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit DINOv3ViTModel‎ does not have layer attribute, DINOv3ViTEncoder has. We have to access through self.model.model --- trellis2/modules/image_feature_extractor.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/trellis2/modules/image_feature_extractor.py b/trellis2/modules/image_feature_extractor.py index c3cb515a..987a60c0 100644 --- a/trellis2/modules/image_feature_extractor.py +++ b/trellis2/modules/image_feature_extractor.py @@ -83,7 +83,7 @@ def extract_features(self, image: torch.Tensor) -> torch.Tensor: hidden_states = self.model.embeddings(image, bool_masked_pos=None) position_embeddings = self.model.rope_embeddings(image) - for i, layer_module in enumerate(self.model.layer): + for i, layer_module in enumerate(self.model.model.layer): hidden_states = layer_module( hidden_states, position_embeddings=position_embeddings,