fc4d6ebf39
* Initial support for hidream. Lora keys likely need work * Fix saving for hidream-o1 * Remove dependence on flash attention for hidream o1 * Fix gradient checkpointing for hidream o1 * A lot of fixes for hidream. Handle loading and saving as comfy model. * Omit layers not used in comfy. Fix issue with lora loading keys in comfy * Version bumpo
4 lines
130 B
Python
4 lines
130 B
Python
from .hidream_model import HidreamModel
|
|
from .hidream_e1_model import HidreamE1Model
|
|
from .hidream_o1_model import HidreamO1Model
|