flex_model.distributed.BaseFunctionStrategy

class flex_model.distributed.BaseFunctionStrategy(user_func: Callable)

Defines an editing function execution strategy. Can validate an editing function (ie. check breakpoints, etc.). Determines which tensors have the editing function applied.

__init__(user_func: Callable)

Methods

__init__(user_func)
execute(*args, **kwargs)
initialize(func)