def multistream_mode() -> int:
    """
    Multi-stream mode. 0 - no multistream. 1 - simple multistream.
    Default: 0.
    """

    # temporarily override it in order to test
    return int(os.getenv("AIT_MULTISTREAM_MODE", "0"))
Multi-stream mode. 0 - no multistream. 1 - simple multistream.
Default: 0.
python/aitemplate/utils/environ.py