diffusion

OptimalScale · Apr 16, 2023 · 68f0980 · 68f0980
1 parent 6765c9c
commit 68f0980
Show file tree

Hide file tree

Showing 3 changed files with 591 additions and 65 deletions.
diff --git a/experimental/RAFT-diffusion/README.md b/experimental/RAFT-diffusion/README.md
@@ -0,0 +1,105 @@
+# RAFT-Diffusion
+
+
+In this folder, we provide an example to show that how does RAFT work on diffusion models. We will also include these script into LMFlow framework into the LMFlow APIs in the future.
+
+
+The requirements are shown below.
+```
+accelerate               0.18.0
+asttokens                2.2.1
+backcall                 0.2.0
+bitsandbytes             0.37.2
+certifi                  2022.12.7
+charset-normalizer       3.1.0
+clip                     1.0        
+cmake                    3.26.1
+comm                     0.1.3
+contourpy                1.0.7
+cycler                   0.11.0
+debugpy                  1.6.7
+decorator                5.1.1
+diffusers                0.14.0
+executing                1.2.0
+filelock                 3.11.0
+fonttools                4.39.3
+ftfy                     6.1.1
+huggingface-hub          0.13.4
+idna                     3.4
+importlib-metadata       6.2.0
+importlib-resources      5.12.0
+ipykernel                6.22.0
+ipython                  8.12.0
+jedi                     0.18.2
+Jinja2                   3.1.2
+jupyter_client           8.1.0
+jupyter_core             5.3.0
+kiwisolver               1.4.4
+lit                      16.0.0
+MarkupSafe               2.1.2
+matplotlib               3.7.1
+matplotlib-inline        0.1.6
+mpmath                   1.3.0
+mypy-extensions          1.0.0
+nest-asyncio             1.5.6
+networkx                 3.1
+numpy                    1.24.2
+nvidia-cublas-cu11       11.10.3.66
+nvidia-cuda-cupti-cu11   11.7.101
+nvidia-cuda-nvrtc-cu11   11.7.99
+nvidia-cuda-runtime-cu11 11.7.99
+nvidia-cudnn-cu11        8.5.0.96
+nvidia-cufft-cu11        10.9.0.58
+nvidia-curand-cu11       10.2.10.91
+nvidia-cusolver-cu11     11.4.0.1
+nvidia-cusparse-cu11     11.7.4.91
+nvidia-nccl-cu11         2.14.3
+nvidia-nvtx-cu11         11.7.91
+open-clip-torch          2.16.0
+packaging                23.0
+pandas                   2.0.0
+parso                    0.8.3
+pexpect                  4.8.0
+pickleshare              0.7.5
+Pillow                   9.5.0
+pip                      23.0.1
+platformdirs             3.2.0
+prompt-toolkit           3.0.38
+protobuf                 3.20.3
+psutil                   5.9.4
+ptyprocess               0.7.0
+pure-eval                0.2.2
+Pygments                 2.14.0
+pyparsing                3.0.9
+pyre-extensions          0.0.23
+python-dateutil          2.8.2
+pytz                     2023.3
+PyYAML                   6.0
+pyzmq                    25.0.2
+regex                    2023.3.23
+requests                 2.28.2
+sentencepiece            0.1.97
+setuptools               65.6.3
+six                      1.16.0
+stack-data               0.6.2
+sympy                    1.11.1
+timm                     0.6.13
+tokenizers               0.13.3
+torch                    2.0.0
+torchvision              0.15.1
+tornado                  6.2
+tqdm                     4.65.0
+traitlets                5.9.0
+transformers             4.27.4
+triton                   2.0.0
+typing_extensions        4.5.0
+typing-inspect           0.8.0
+tzdata                   2023.3
+urllib3                  1.26.15
+wcwidth                  0.2.6
+wheel                    0.38.4
+xformers                 0.0.18
+zipp                     3.15.0
+```
+
+We will also add a COLAB link for convenience.