environment.yml

channels:
  - pytorch
  - nvidia
dependencies:
  - python=3.11.3
  - pytorch=2.0.1
  - pytorch-cuda=11.7
  - torchvision=0.15.2
  - pip:
      - accelerate==0.19.0
      - diffusers[torch]==0.16.0
      - einops==0.6.1
      - ffmpeg==1.4
      - pandas==2.0.1
      - tqdm==4.65.0
      - transformers==4.28.1
      - xformers==0.0.20
      - rotary_embedding_torch
      - gradio==4.3.0