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