mirror of
https://github.com/hpcaitech/ColossalAI.git
synced 2026-02-21 06:22:09 +00:00
50cdf5430e
* [builder] builder for scaled_upper_triang_masked_softmax * add missing files * fix a bug * polish code * [example] diffusion install from docker
18 lines
285 B
Plaintext
18 lines
285 B
Plaintext
albumentations==1.3.0
|
|
opencv-python==4.6.0
|
|
pudb==2019.2
|
|
prefetch_generator
|
|
imageio==2.9.0
|
|
imageio-ffmpeg==0.4.2
|
|
torchmetrics==0.6
|
|
omegaconf==2.1.1
|
|
test-tube>=0.7.5
|
|
streamlit>=0.73.1
|
|
einops==0.3.0
|
|
transformers==4.19.2
|
|
webdataset==0.2.5
|
|
open-clip-torch==2.7.0
|
|
gradio==3.11
|
|
datasets
|
|
-e .
|