From b76f2e46639fd608f80d02b2df124f98cfdeee60 Mon Sep 17 00:00:00 2001
From: Stefan Imhoff <stefan@imhoff.name>
Date: Sat, 8 Oct 2022 11:44:23 +0200
Subject: [PATCH] chore(python): add more packages

---
 setup/python.sh | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/setup/python.sh b/setup/python.sh
index 5d3acf2..2a65d1e 100755
--- a/setup/python.sh
+++ b/setup/python.sh
@@ -11,11 +11,13 @@ eval "$(pyenv virtualenv-init -)"
 
 pyenv install 3.10.1
 
+pip install Pillow
 pip install autopep8
 pip install awscli
 pip install black
 pip install certifi
 pip install codespell
+pip install diffusers==0.3.0
 pip install fawkes
 pip install flake8
 pip install git+https://github.com/openai/whisper.git
@@ -26,8 +28,12 @@ pip install jupyterlab
 pip install mutagen
 pip install neovim
 pip install notebook
+pip install numpy
 pip install proselint
 pip install pycryptodomex
+pip install torch
+pip install tqdm
+pip install transformers
 pip install vim-vint
 pip install voila
 pip install websockets