text-generation-inference

29b8d19c - fix: return the out tensor rather then the functions return value (#2361)