jbilcke-hf's picture
jbilcke-hf HF Staff
upgrading finetrainers (and losing my extra code + improvements)
80ebcb3
raw
history blame
148 Bytes
DIFFUSERS_TRANSFORMER_BLOCK_NAMES = [
"transformer_blocks",
"single_transformer_blocks",
"temporal_transformer_blocks",
"blocks",
]