diff --git a/pyproject.toml b/pyproject.toml index 5626c3d8..736f546a 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -108,7 +108,7 @@ authors = [ ] license = {file = "LICENSE"} readme = "README.md" -requires-python = ">=3.10,<3.13" +requires-python = ">=3.10,<3.14" keywords = ["AI", "machine learning", "model optimization", "pruning"] classifiers = [ "Development Status :: 4 - Beta", @@ -161,7 +161,6 @@ dependencies = [ "peft>=0.18.0", "trl<=0.21.0", "termcolor==2.3.0", - "realesrgan" ] [project.optional-dependencies] @@ -193,6 +192,9 @@ awq = [ "llmcompressor>=0.9", "torch>=2.9.0" ] +upscale = [ + "realesrgan", +] full = [ "pruna[stable-fast]", ] diff --git a/src/pruna/algorithms/upscale.py b/src/pruna/algorithms/upscale.py index da28224c..b3a848a2 100644 --- a/src/pruna/algorithms/upscale.py +++ b/src/pruna/algorithms/upscale.py @@ -80,6 +80,7 @@ class RealESRGAN(PrunaAlgorithmBase): "ring_attn", "hyper", ] + required_install: str = "``pip install pruna[upscale]``" def get_hyperparameters(self) -> list: """