environment.yaml

name: s3diff
channels:
  - pytorch
  - defaults
dependencies:
  - python=3.10
  - pip:
      - einops>=0.6.1
      - numpy>=1.24.4
      - open-clip-torch>=2.20.0
      - opencv-python==4.6.0.66
      - pillow>=9.5.0
      - scipy==1.11.1
      - timm>=0.9.2
      - tokenizers
      - torch>=2.0.1

      - torchaudio>=2.0.2
      - torchdata==0.6.1
      - torchmetrics>=1.0.1
      - torchvision>=0.15.2

      - tqdm>=4.65.0
      - transformers==4.35.2
      - triton==2.0.0
      - urllib3<1.27,>=1.25.4
      - xformers>=0.0.20
      - streamlit-keyup==0.2.0
      - lpips
      - peft
      - pyiqa
      - omegaconf
      - dominate
      - diffusers==0.25.1
      - gradio==5.4.0