rct_model / test_pipeline.py

Commit History

Use float16 for inference and float32 for training
42f8b67

frutiemax commited on

Use SGD and text encoder/tokenizer
6fa0b52

frutiemax commited on

Register modules in pipeline
5961f34

frutiemax commited on

Fix other things...
a150f0f

frutiemax commited on

Use VAE to speed up inference
054faf7

frutiemax commited on

Use ConditionalUnetModel
88deab4

frutiemax commited on

Training code
d751051

frutiemax commited on

Push initial pipeline
2a7e546

frutiemax commited on