Taken from https://github.com/tracel-ai/models/commit/42a453dfec66d20ff66d35801dca10d9a998182f
