mayukhdeb 4385a55a23 keep all optimizers in one place

2024-04-15 00:25:26 -07:00

..

utils

add token warmup

2024-04-08 19:36:54 -07:00

config.py

try to debug bug issue

2024-04-14 11:20:47 -07:00

dataset.py

try to debug bug issue

2024-04-14 11:07:49 -07:00

embedding_handler.py

remove ti from dataloader, simplify

2024-04-08 23:43:33 -07:00

inference.py

unet full finetuning + checkpointing + eval (commented out gradient norm logging temporarily)

2024-04-11 06:43:01 -07:00

lora.py

unet full finetuning + checkpointing + eval (commented out gradient norm logging temporarily)

2024-04-11 06:43:01 -07:00

loss.py

update todos

2024-04-09 00:17:01 -07:00

models.py

update defaults

2024-04-14 08:09:33 -07:00

optimizer.py

keep all optimizers in one place

2024-04-15 00:25:26 -07:00

preprocess.py

fix resolution bug

2024-04-09 17:53:07 -07:00