From 4d0506d73d153ab6e165302be2a2425153574c82 Mon Sep 17 00:00:00 2001
From: Benedikt Blumenstiel <benedikt.blumenstiel@ibm.com>
Date: Wed, 15 Jan 2025 15:07:23 +0100
Subject: [PATCH] Fix timm config loading for prithvi
Signed-off-by: Benedikt Blumenstiel <benedikt.blumenstiel@ibm.com>
---
terratorch/models/backbones/prithvi_vit.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/terratorch/models/backbones/prithvi_vit.py b/terratorch/models/backbones/prithvi_vit.py
index 136c6513..9918d5f0 100644
--- a/terratorch/models/backbones/prithvi_vit.py
+++ b/terratorch/models/backbones/prithvi_vit.py
@@ -208,7 +208,7 @@ def checkpoint_filter_wrapper_fn(state_dict, model):
f"(pretrained models: {default_cfgs.keys()})")
# Load pre-trained config from hf
try:
- model_args, _ = load_model_config_from_hf(default_cfgs[variant].default.hf_hub_id)
+ model_args = load_model_config_from_hf(default_cfgs[variant].default.hf_hub_id)[0]
model_args.update(kwargs)
except:
logger.warning(f"No pretrained configuration was found on HuggingFace for the model {variant}."