Add lion lib and speed test

Train_JupyterLab.ipynb

@@ -32,6 +32,28 @@
     "Come visit us at [EveryDream Discord](https://discord.gg/uheqxU6sXN)"
    ]
   },
+  {
+   "cell_type": "markdown",
+   "id": "ffff47f7",
+   "metadata": {},
+   "source": [
+    "## Optional Speed Test\n",
+    "If all goes well you may find yourself downloading (or pushing to the cloud) 2-8GB of model data per saved checkpoint. Make sure your pod is not a dud. ~1000Mbit/s up/dn is probably good, though the location of the pod also makes a difference.\n"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "934ba107",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "import speedtest\n",
+    "st = speedtest.Speedtest()\n",
+    "print(f\"Your download speed: {round(st.download() / 1000 / 1000, 1)} Mbit/s\")\n",
+    "print(f\"Your upload speed: {round(st.upload() / 1000 / 1000, 1)} Mbit/s\")"
+   ]
+  },
   {
    "cell_type": "markdown",
    "id": "7c73894e-3b5e-4268-9f83-ed89bd4569f2",

docker/Dockerfile

@@ -74,6 +74,7 @@ RUN echo "source ${VIRTUAL_ENV}/bin/activate" >> /root/.bashrc
 ENV LD_LIBRARY_PATH="/usr/local/cuda-11.8/targets/x86_64-linux/lib/"
 RUN ln /usr/local/cuda/targets/x86_64-linux/lib/libcudart.so.11.8.89 /usr/local/cuda-11.8/targets/x86_64-linux/lib/libcudart.so
 RUN ln /usr/local/cuda/targets/x86_64-linux/lib/libnvrtc.so.11.8.89 /usr/local/cuda-11.8/targets/x86_64-linux/lib/libnvrtc.so
+
 ADD requirements-runtime.txt /
 RUN pip install --no-cache-dir -r requirements-runtime.txt
 
@@ -81,7 +82,6 @@ WORKDIR /workspace
 RUN git clone https://github.com/victorchall/EveryDream2trainer
 
 WORKDIR /workspace/EveryDream2trainer
-# RUN git checkout torch2
 RUN python utils/get_yamls.py && \
     mkdir -p logs && mkdir -p input
 

docker/requirements-runtime.txt

@@ -6,8 +6,10 @@ ipyevents
 ipywidgets
 jupyter-archive
 jupyterlab
+lion-pytorch
 piexif==1.1.3
 pyfakefs
 pynvml==11.5.0
+speedtest-cli
 tensorboard==2.12.0
 wandb