diff --git a/README.md b/README.md index fea2b833..2552d95e 100644 --- a/README.md +++ b/README.md @@ -46,12 +46,12 @@ Pruna is a model optimization framework built for developers, enabling you to de The toolkit is designed with simplicity in mind - requiring just a few lines of code to optimize your models. It supports various model types including LLMs, Diffusion and Flow Matching Models, Vision Transformers, Speech Recognition Models and more. - + ## Pruna Cool Installation @@ -126,7 +126,7 @@ eval_agent.evaluate(smashed_model) ``` This was the minimal example, but you are looking for the maximal example? You can check out our [documentation][documentation] for an overview of all supported [algorithms][docs-algorithms] as well as our tutorials for more use-cases and examples. - + ## Pruna Cool Algorithm Overview @@ -158,13 +158,9 @@ Since Pruna offers a broad range of optimization algorithms, the following table | `batcher` | Groups multiple inputs together to be processed simultaneously, improving computational efficiency and reducing processing time. | ✅ | ❌ | ➖ | | `cacher` | Stores intermediate results of computations to speed up subsequent operations. | ✅ | ➖ | ➖ | | `compiler` | Optimises the model with instructions for specific hardware. | ✅ | ➖ | ➖ | -| `distiller` | Trains a smaller, simpler model to mimic a larger, more complex model. | ✅ | ✅ | ❌ | | `quantizer` | Reduces the precision of weights and activations, lowering memory requirements. | ✅ | ✅ | ❌ | | `pruner` | Removes less important or redundant connections and neurons, resulting in a sparser, more efficient network. | ✅ | ✅ | ❌ | -| `recoverer` | Restores the performance of a model after compression. | ➖ | ➖ | ✅ | | `factorizer` | Factorization batches several small matrix multiplications into one large fused operation. | ✅ | ➖ | ➖ | -| `enhancer` | Enhances the model output by applying post-processing algorithms such as denoising or upscaling. | ❌ | ➖ | ✅ | -| `distributer` | Distributes the inference, the model or certain calculations across multiple devices. | ✅ | ❌ | ➖ | | `kernel` | Kernels are specialized GPU routines that speed up parts of the computation. | ✅ | ➖ | ➖ | ✅ (improves), ➖ (approx. the same), ❌ (worsens)