text-generation-inference/server/text_generation/__init__.py