Refine generation code for trained unet checkpoint - Githubissues

Nota-NetsPresso / BK-SDM

A Compressed Stable Diffusion for Efficient Text-to-Image Generation [ECCV'24]

Other

238 stars 16 forks source link

Refine generation code for trained unet checkpoint #18

Closed bokyeong1015 closed 1 year ago

bokyeong1015 commented 1 year ago

16 for trained unet
- remove use_auth_token=True in StableDiffusionPipeline.from_pretrained
- remove src/generate_with_trained_unet.py (integrated into src/generate.py)
- add scripts/generate_with_trained_unet.sh
- disable NSFW filter for benchmark with diffusers >=0.19
Refine readme
- introduce inference process with trained unet
- reorganize the order of sections: training -> eval
- shorten sentences & add some hyperlinks