Speed_Read_AI / environment.yml
circulartext's picture
Create environment.yml
2c8c0a9
raw
history blame
388 Bytes
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