Hugging Face's logo

Spaces:
damian0815
/
Erasing-Concepts-In-Diffusion
Runtime error

App Files Community
Erasing-Concepts-In-Diffusion / train.py

Commit History

add tensorboard, validation, sample output
50b9662

Damian Stewart commited on

save every N steps and loss logging
6067469

Damian Stewart commited on

fix seed and export
c8aa68b

Damian Stewart commited on

add train seed
94be4c7

Damian Stewart commited on

typo
b58675c

Damian Stewart commited on

actually use AMP=3x speedup
ab11bdd

Damian Stewart commited on

wip adding AMP and xformers to training code path
fc73e59

Damian Stewart commited on

allow different resolutions for w/h
d8ffb68

Damian Stewart commited on

provide empty negative prompt when training
209d166

Damian Stewart commited on

support for different base models
0002379

Damian Stewart commited on

Updated diffuser to use seperate generator
7c89716

JadenFK commited on

Merge branch 'main' of https://huggingface.co/spaces/baulab/Erasing-Concepts-In-Diffusion into main
87cd610

JadenFK commited on

Memory saving
7021212

JadenFK commited on

More refactoring
81ccbca

JadenFK commited on