Faster assisted generation support for Intel Gaudi