Transition to Pruna Pro

pruna_pro is our premium offering that provides advanced compression algorithms and features to help you achieve better model compression results. This guide will help you get started with pruna_pro and explain how to transition from the open-source pruna. Make sure to check out the installation instructions in /setup/pip first.

If you’re already familiar with pruna, transitioning to pruna_pro is straightforward. The main interfaces remain the same, with just a few adjustments:

Installing Pruna Pro

If you want to install the Pro version of pruna_pro, you can do so with the following command:

pip install pruna_pro==|version|

Similarly to pruna, you can install specific algorithm dependencies or the full version of pruna_pro.

Using the Pruna Pro Interface

  1. Get your pruna_pro token.

We will release an automatic payment system in the coming release, for now, here’s how to get your token:

  • Join our Discord server

  • Go to the #pruna-pro channel

  • Request a token in the channel by sending a quick message

  1. Import from pruna_pro instead of pruna:

    from pruna_pro import smash  # instead of: from pruna import smash

  2. Pass your pruna_pro token to the smash function:

    from pruna_pro import smash

smash(model, smash_config, token='<your_pruna_pro_token>') # add your token here

  1. Use the same familiar workflow with enhanced capabilities:

    • The algorithm documentation applies to both versions

    • Configuration parameters work the same way

    • All pruna features are available in pruna_pro, with additional advanced options

Key Features in pruna_pro

pruna_pro includes all features from the open-source version plus premium capabilities:

  • Advanced compression algorithms

  • Optimization agent that automatically finds the best compression configuration for your model

  • Quality recovery algorithms that can recover the accuracy of a smashed model

  • Priority support

For a complete feature comparison and pricing details, visit our pricing page.

Note

The documentation for core features applies to both pruna and pruna_pro. When a feature is Pro-only, it will be clearly marked with a “(Pro)” badge.