nebuly-ai / optimate

A collection of libraries to optimise AI model performances

Home Page:https://www.nebuly.com/

Repository from Github https://github.comnebuly-ai/optimateRepository from Github https://github.comnebuly-ai/optimate

[Speedster] Make Speedster optimize_model() return InferenceLearner also for StableDiffusion models

Telemaco019 opened this issue · comments

Speedster optimize_model function should return an object of type BaseInferenceLearner. However when optimizing Stable Diffusion models, the function returns an object of type StableDiffusionPipeline.

It would be useful if the function returned an Inference Learner also for Stable Diffusion to have a common API and expose utility methods such as save() and get_size() on the optimized models.