APIs

API reference: pe.api package

pe.api.api.API is responsible for implementing the foundation model APIs. It has the following key methods:

• pe.api.api.API.random_api(): Randomly generates the synthetic samples for the initial samples of the Private Evolution algorithm.

• pe.api.api.API.variation_api(): Generates the variations of the given synthetic samples for the initial or the next Private Evolution iteration.

Available APIs

Currently, the following APIs are implemented:

• Images

  • pe.api.image.stable_diffusion_api.StableDiffusion: The APIs of Stable Diffusion.

  • pe.api.image.improved_diffusion_api.ImprovedDiffusion: The APIs of the improved diffusion model.

• Text

  • Coming soon!

Adding Your Own APIs

To add your own APIs, you need to create a class that inherits from pe.api.api.API and implements the pe.api.api.API.random_api() and pe.api.api.API.variation_api() methods.