fix: add realesrgan as a dependency to fix basicsr missing problem (#569)

* fix: add realesrgan as a dependency to fix basicsr missing problem

* ci: update gliner python dependency

* feat: super duper new features

* fix: remove feature time

* fix: add protection around gliner import in torch structured

---------

Co-authored-by: Gaspar Rochette <gaspar.rochette@pruna.ai>

Author: begumcig

pyproject.toml

@@ -143,7 +143,7 @@ dependencies = [
     "whisper-s2t==1.3.1",
     "hqq==0.2.7.post1",
     "torchao>=0.12.0,<0.16.0", # 0.16.0 breaks diffusers 0.36.0, torch+torch: https://github.com/pytorch/ao/issues/2919#issue-3375688762
-    "gliner; python_version >= '3.10'",
+    "gliner; python_version >= '3.11'",
     "piq",
     "opencv-python",
     "kernels",
@@ -153,6 +153,7 @@ dependencies = [
     "peft>=0.18.0",
     "trl<=0.21.0",
     "termcolor==2.3.0",
+    "realesrgan"
 ]
 
 [project.optional-dependencies]

src/pruna/algorithms/torch_structured.py

@@ -149,7 +149,7 @@ def model_check_fn(self, model: Any) -> bool:
             return True
         if isinstance(model, imported_modules["torchvision"].models.resnet.ResNet):
             return True
-        if isinstance(model, imported_modules["GLiNER"]):
+        if imported_modules["GLiNER"] is not None and isinstance(model, imported_modules["GLiNER"]):
             return True
         return isinstance(model, imported_modules["timm"].models.resnet.ResNet)
 
@@ -257,7 +257,6 @@ def import_algorithm_packages(self) -> Dict[str, Any]:
             import timm
             import torch_pruning as tp
             import torchvision
-            from gliner import GLiNER
             from timm.models.mvitv2 import MultiScaleAttention
             from timm.models.mvitv2 import MultiScaleVit as MViT
             from transformers.models.llama.modeling_llama import LlamaForCausalLM as Llama
@@ -266,6 +265,10 @@ def import_algorithm_packages(self) -> Dict[str, Any]:
         except ImportError:
             pruna_logger.error("TorchStructuredPruner: You need the GPU version of Pruna (timm, torchvision).")
             raise
+        try:
+            from gliner import GLiNER
+        except ImportError:  # onnxruntime doesn't have python3.10- wheels
+            GLiNER = None  # noqa: N806  # type: ignore[assignment]
         return dict(
             timm=timm,
             torchvision=torchvision,