Ryan Dick 03cf953398 First pass of utility function to infer the FluxParams from a state dict.

2024-10-10 07:59:29 -04:00

..

__init__.py

Copy ControlNetFlux model from 47495425db/src/flux/controlnet.py.

2024-10-10 07:59:29 -04:00

diffusers_controlnet_flux.py

(minor) rename other_forward() -> forward()

2024-10-10 07:59:29 -04:00

state_dict_utils.py

First pass of utility function to infer the FluxParams from a state dict.

2024-10-10 07:59:29 -04:00

xlabs_controlnet_flux.py

Improve typing of zero_module().

2024-10-10 07:59:29 -04:00

zero_module.py

Improve typing of zero_module().

2024-10-10 07:59:29 -04:00