frutiemax
/

rct_model

Model card Files Files and versions Community

rct_model / train_model.py

Commit History

Switch to stabilityai/sd-vae-ft-mse

2104644

frutiemax commited on Sep 23, 2023

Use VAE to speed up inference

054faf7

frutiemax commited on Sep 23, 2023

Revert to not using accelerated

21640c8

frutiemax commited on Sep 23, 2023

Push unet and scheduler outside of pipeline

a4c8091

frutiemax commited on Sep 23, 2023

Use accelerate

f6f5f48

frutiemax commited on Sep 23, 2023

Fix batch size

1618027

frutiemax commited on Sep 23, 2023

Fix training

04d70cd

frutiemax commited on Sep 23, 2023

Use ConditionalUnetModel

88deab4

frutiemax commited on Sep 23, 2023

Training code

d751051

frutiemax commited on Sep 22, 2023