SwinIR VS dalle-playground

We’re referencing our three models directly as git repos (because they’re not available as packages in PyPI) in model_requirements.txt, but we had to make a few changes to make these repos work as pip packages. Pip looks for a setup.py file in the git repo to figure out which files from the repo need to be installed into the environment, as well as what the dependencies of that repo are. GLID-3-xl and latent-diffusion (another diffusion model that GLID-3-xl depends on) had setup.py files that needed tweaks to include all of the code needed to run the models. SwinIR didn’t have a setup.py file at all, so we added one. Finally, all of these setup.py files needed additional dependencies, which we just added to the model_requirements.txt file.

Built on dalle-playground because it is simple and I like it.

[1] https://github.com/saharmor/dalle-playground

If you decide to abandon the idea of running locally and want to run in the cloud instead, https://github.com/saharmor/dalle-playground has a Google Colab notebook that's relatively easy to run (although Google Colab's free tier is relatively limited).

Next, we want the code in the https://github.com/hrichardlee/dalle-playground repo, and we want to construct a pip environment from the backend/requirements.txt file in that repo. We were almost able to use the saharmor/dalle-playground repo as-is, but we had to make one change to add the jax[cuda] package to the requirements.txt file. In case you haven’t seen jax before, jax is a machine-learning library from Google, roughly equivalent to Tensorflow or PyTorch. It combines Autograd for automatic differentiation and XLA (accelerated linear algebra) for JIT-compiling numpy-like code for Google’s TPUs or Nvidia’s CUDA API for GPUs. The CUDA support requires explicitly selecting the [cuda] option when we install the package.

Is their any way to run DALLE-2 inside of a docker container similarly to this DALLE-PLAYGROUND repo on github?

I know there are open source projects like dalle-playground. Would it literally be enough to host this app on an EC2 instance with the mini model?