Unionを受け取る関数の返り値の型を引数の型によって変えたいときってどう書けばいいの？

type F<T> = (() => void) | ((x: T) => void);

const wrap = <T>(f: F<T>) => ??? ;

const a: () => void = wrap( () => {} );
const b: (x: string) => void = wrap( (x: string) => {} );