environment.yml

name: inferbench
channels:
  - conda-forge
  - defaults
dependencies:
  - python=3.12
  - numpy
  - opencv
  - pillow
  - python-dotenv
  - requests
  - tqdm
  - pip
  - pip:
      - datasets==3.6.0
      - fal-client>=0.5.9
      - hpsv2>=1.2.0
      - huggingface-hub>=0.30.2
      - image-reward>=1.5
      - replicate>=1.0.4
      - t2v-metrics>=1.2
      - together>=1.5.5
      - torch>=2.7.0
      - torchmetrics>=1.7.1
      - clip
      - diffusers<=0.31
      - piq>=0.8.0