environment_osx-arm64.yml

name: dreem

channels:
  - pytorch
  - conda-forge
  - defaults

dependencies:
  - python=3.11
  - pytorch
  - torchvision
  - lightning
  - imageio
  - av
  - ffmpeg
  - albumentations
  - matplotlib
  - pip
  - pip:
    - "--editable=.[dev]"
    - timm