diff --git a/trellis2/trainers/flow_matching/mixins/image_conditioned_proj.py b/trellis2/trainers/flow_matching/mixins/image_conditioned_proj.py index bb7fa84..d03d83e 100644 --- a/trellis2/trainers/flow_matching/mixins/image_conditioned_proj.py +++ b/trellis2/trainers/flow_matching/mixins/image_conditioned_proj.py @@ -441,7 +441,7 @@ class DinoV3ProjFeatureExtractor(nn.Module): self.naf_model = torch.hub.load( cached_repo, "naf", pretrained=True, device=device, source="local", trust_repo=True ) - elif self.naf_download_if_missing: + else: self.naf_model = torch.hub.load( "valeoai/NAF", "naf", pretrained=True, device=device, trust_repo=True )