search

alibaba / EasyNLP

EasyNLP: A Comprehensive and Easy-to-use NLP Toolkit
Apache License 2.0
2.03k stars 250 forks source link

latent diffusion training code #273

Closed Bingyan-Liu closed 1 year ago

CLAassistant commented 1 year ago

CLA assistant check
All committers have signed the CLA.

Bingyan-Liu commented 1 year ago

add txt-img data pairs reading code in easynlp/appzoo/latent_diffusion/data. add ldm_utils.py and lr_scheduler.py in easynlp/utils. add original class LatentDiffusion(DDPM) in easynlp/modelzoo/models/latent_diffusion/ddpm.py. add readme.md and update main.py for finetune latent diffusion in examples/latent_diffusion.