Skip to content

mles-02/diffusion-optimize

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

23 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Finetune, Optimize and Deploy with Stable Diffusion

We introduce a complete pipeline from finetune to model optimization, finally deploying to the Cloud with a text-to-image Stable Diffusion.

Authors:

Advisors:

📝 Documentation

Currently, we investigate methods for finetuning Stable Diffusion with 🤗 diffuers.

Due to hardware limitations, our finetune model cannot achieve the best results and is only to demonstrate the feasibility of these methods.

We have researched and applied the optimization methods from 🤗 diffuers and 🤗 optimum.

We provided instructions to deploy Stable Diffusion pipeline on Google Cloud. Note that we are currently only deploying on CPU. Work with GPU is plan for future works.

✔️ TODO

  • Build UI for Text-to-Image
  • Complete deployment with Triton Inference Server
  • Investigate and write reports for optimization methods in GPU
  • Evaluation of model quality

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Contributors 3

  •  
  •  
  •