frutiemax
/

rct_model

Model card Files Files and versions Community

rct_model / rct_diffusion_pipeline.py

Commit History

Use ImageVaeProcessor

aa6b13c

frutiemax commited on Oct 2, 2023

Dont decode in the training phase

3ab0859

frutiemax commited on Oct 1, 2023

Use vae for encoding and decoding for training

9bde8da

frutiemax commited on Oct 1, 2023

Use float16 for inference and float32 for training

42f8b67

frutiemax commited on Oct 1, 2023

Use SGD and text encoder/tokenizer

6fa0b52

frutiemax commited on Sep 24, 2023

Register modules in pipeline

5961f34

frutiemax commited on Sep 23, 2023

Fix other things...

a150f0f

frutiemax commited on Sep 23, 2023

Fix pipeline for new latent and sample sizes

4f25fc2

frutiemax commited on Sep 23, 2023

Use VAE to speed up inference

054faf7

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 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

Push initial pipeline

2a7e546

frutiemax commited on Sep 20, 2023