mayukhdeb cde4432588 cleaner checkpoint saving WIP

2024-04-17 04:56:34 -07:00

..

utils

add token warmup

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

config.py

offload textual inversion stuff

2024-04-16 05:26:35 -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

cleaner checkpoint saving WIP

2024-04-17 04:56:34 -07:00

loss.py

fix textual_inversion bug

2024-04-16 03:22:10 +02:00

models.py

remove trainable embeddings from print

2024-04-15 08:36:18 -07:00

optimizer.py

fix save/load bug

2024-04-16 21:57:10 +02:00

preprocess.py

fix textual_inversion bug

2024-04-16 03:22:10 +02:00