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.

To get a new token, please visit our pricing page and select Buy Pro. You will be able to choose between a usage-based model or buying credits.

  • Usage-based: $0.40/hour, as much usage as you need billed at the end of the month

  • Credit-based: you purchase credits in multiples of $200 ($200 = 500 hours of inference time)

After selecting your preferred mode, you will be redirected to Stripe. If all goes through, you will receive an email from us with the token. Should you have any issues, please contact us at support@pruna.ai; we’ll make sure to solve any problems or answer any questions.

  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.