brian6091

A100-SXM4-40GB * GPU=16168/40536MiB * 1.23s/it training, 5.83 it/s inference * BATCH_SIZE=4 * TRAIN_TEXT_ENCODER * USE_8BIT_ADAM * FP16 * GRADIENT_CHECKPOINTING * GRADIENT_ACCUMULATION_STEPS=1 * USE_EMA=False * RESOLUTION=512 * No errors or warnings...

Setting feature_extractor=None does not seem to affect inference for SD v1-5