Add some llm setup

diff --git a/invoke/setup-llm/serve.sh b/invoke/setup-llm/serve.sh
new file mode 100755 (executable)
index 0000000..2c8ea6f
--- /dev/null
+++ b/invoke/setup-llm/serve.sh
@@ -0,0 +1,2 @@
+source $HOME/envs/sd/bin/activate
+$HOME/src/stable-diffusion-webui/webui.sh --listen

diff --git a/invoke/setup-llm/setup-sd.sh b/invoke/setup-llm/setup-sd.sh
new file mode 100755 (executable)
index 0000000..7b7df2c
--- /dev/null
+++ b/invoke/setup-llm/setup-sd.sh
@@ -0,0 +1,8 @@
+# Stable Diffusion setup
+mkdir -p $HOME/envs
+python -m venv $HOME/envs/uv
+# Stable Diffusion requires Python 3.11
+# --seed adds pip, which SD needs for setup
+cd $HOME/envs;$HOME/envs/uv/bin/uv venv --seed --python 3.11 sd
+mkdir -p $HOME/src
+cd $HOME/src; git clone https://github.com/AUTOMATIC1111/stable-diffusion-webui

diff --git a/invoke/setup-llm/tasks.py b/invoke/setup-llm/tasks.py
new file mode 100644 (file)
index 0000000..926fd98
--- /dev/null
+++ b/invoke/setup-llm/tasks.py
@@ -0,0 +1,5 @@
+
+
+@task
+def check_nvidia(c):
+    c.run('nvidia-smi')