enforce einops 0.6.1 or greater and make sure torch compile friendly

import torch

from packaging import version

if version.parse(torch.__version__) >= version.parse('2.0.0'):

    from einops._torch_specific import allow_ops_in_compiled_graph

    allow_ops_in_compiled_graph()

from audiolm_pytorch.audiolm_pytorch import AudioLM

from audiolm_pytorch.soundstream import SoundStream, AudioLMSoundStream, MusicLMSoundStream

from audiolm_pytorch.encodec import EncodecWrapper

__version__ = '0.26.6'

__version__ = '0.26.8'

  install_requires=[

    'accelerate',

    'beartype',

    'einops>=0.6',

    'einops>=0.6.1',

    'ema-pytorch>=0.2.2',

    'encodec',

    'fairseq',