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|
pip install pruna_pro==|version| --extra-index-url https://download.pytorch.org/whl/cpu
Similarly to pruna, you can install specific algorithm dependencies or the full
version of pruna_pro.
Using the Pruna Pro Interface
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.
Import from
pruna_pro
instead ofpruna
:from pruna_pro import smash # instead of: from pruna import smash
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
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.