Fixed NAF Downloading for Pixal3D

This commit is contained in:
Bruno Fargnoli
2026-05-26 19:08:41 +02:00
parent a2e4dec572
commit 0b324cd75d
@@ -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
)