[train_text_to_image] allow using non-ema weights for training #1834
allow using non-ema weights for training
32c7cdee
Apply suggestions from code review
a706c96e
address more review comment
5918f669
reorganise a few lines
103eacab
Merge branch 'main' into finetune-non-ema
6f8654db
always pad text to max_length to match original training
5c8f0699
ifx collate_fn
008014e4
Merge branch 'main' into finetune-non-ema
80b5c190
Merge branch 'main' into finetune-non-ema
0171fcd1
remove unused code
fe51c7c5
don't prepare ema_unet, don't register lr scheduler
50782ebf
style
4079f5b4
assert => ValueError
114969e4
patil-suraj
changed the title [wip][examples/train_text_to_image] allow using non-ema weights for training [train_text_to_image] allow using non-ema weights for training 3 years ago
add allow_tf32
a54a56bd
Merge branch 'main' into finetune-non-ema
aae03ecf
set log level
ad7ebe3e
pcuenca
approved these changes
on 2022-12-30
fix comment
b29ed553
patil-suraj
deleted the finetune-non-ema branch 3 years ago
Assignees
No one assigned
Login to write a write a comment.
Login via GitHub