Brandon Rising 65bb46bcca Rename params for flux and flux vae, add comments explaining use of the config_path in model config

2024-08-26 20:17:50 -04:00

..

modules

Replace swish() with torch.nn.functional.silu(h). They are functionally equivalent, but in my test VAE deconding was ~8% faster after the change.

2024-08-26 20:17:50 -04:00

math.py

Attribute black-forest-labs/flux for much of the flux code

2024-08-26 20:17:50 -04:00

model.py

Attribute black-forest-labs/flux for much of the flux code

2024-08-26 20:17:50 -04:00

sampling.py

Run ruff

2024-08-26 20:17:50 -04:00

util.py

Rename params for flux and flux vae, add comments explaining use of the config_path in model config

2024-08-26 20:17:50 -04:00