Adding new approach

Author: alexanderdann

timm/models/_hub.py

@@ -542,17 +542,17 @@ def _get_safe_alternatives(filename: str) -> Iterable[str]:
         yield filename[:-4] + ".safetensors"
 
 
-def _get_license_from_hf_hub(model_id: str | None, hf_hub_id: str | None) -> str | None:
+def _get_license_from_hf_hub(hf_hub_id: str | None) -> str | None:
     """Retrieve license information for a model from Hugging Face Hub.
 
     Fetches the license field from the model card metadata on Hugging Face Hub
-    for the specified model. Returns None if the model is not found, if
-    huggingface_hub is not installed, or if the model is marked as "untrained".
+    for the specified model. This function is called lazily when the license
+    attribute is accessed on PretrainedCfg objects that don't have an explicit
+    license set.
 
     Args:
-        model_id: The model identifier/name. In the case of None we assume an untrained model.
-        hf_hub_id: The Hugging Face Hub organization/user ID. If it is None,
-            we will return None as we cannot infer the license terms.
+        hf_hub_id: The Hugging Face Hub model ID (e.g., 'organization/model').
+            If None or empty, returns None as license cannot be determined.
 
     Returns:
         The license string in lowercase if found, None otherwise.
@@ -566,17 +566,17 @@ def _get_license_from_hf_hub(model_id: str | None, hf_hub_id: str | None) -> str
         _logger.warning(msg=msg)
         return None
 
-    if not (model_id and hf_hub_id):
+    if hf_hub_id is None or hf_hub_id == "timm/":
         return None
 
-    repo_id: str = hf_hub_id + model_id
-
     try:
-        info = model_info(repo_id=repo_id)
+        info = model_info(repo_id=hf_hub_id)
 
     except RepositoryNotFoundError:
         # TODO: any wish what happens here? @rwightman
-        print(repo_id)
+        return None
+
+    except Exception as _:
         return None
 
     license = info.card_data.get("license").lower() if info.card_data else None

timm/models/_pretrained.py

@@ -58,6 +58,20 @@ class PretrainedCfg:
     def has_weights(self):
         return self.url or self.file or self.hf_hub_id
 
+    def __getattribute__(self, name):
+        if name == 'license': # Intercept license access to set it in case it was not set anywhere else.
+            license_value = super().__getattribute__('license')
+
+            if license_value is None:
+                from ._hub import _get_license_from_hf_hub
+                license_value = _get_license_from_hf_hub(hf_hub_id=self.hf_hub_id)
+
+                self.license = license_value
+
+            return license_value
+
+        return super().__getattribute__(name)
+
     def to_dict(self, remove_source=False, remove_null=True):
         return filter_pretrained_cfg(
             asdict(self),