conditionally load models in tasks based on function signature