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 Install Pruna 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==0.2.10

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

Getting Started with Pruna Pro

  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

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

  3. 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 - Access to cutting-edge compression techniques

  • Optimization agent - Automatically finds the best compression configuration for your model

  • Quality recovery algorithms - Can recover the accuracy of a smashed model

  • Priority support - Get help faster when you need it most

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.

Next Steps