• Pedro Cuenca's avatar
    Return Flax scheduler state (#601) · a9fdb3de
    Pedro Cuenca authored
    * Optionally return state in from_config.
    
    Useful for Flax schedulers.
    
    * has_state is now a property, make check more strict.
    
    I don't check the class is `SchedulerMixin` to prevent circular
    dependencies. It should be enough that the class name starts with "Flax"
    the object declares it "has_state" and the "create_state" exists too.
    
    * Use state in pipeline from_pretrained.
    
    * Make style
    a9fdb3de
configuration_utils.py 20.5 KB