diff --git a/.github/workflows/main.yaml b/.github/workflows/main.yaml
index 0eff66e..ae3e7a2 100644
--- a/.github/workflows/main.yaml
+++ b/.github/workflows/main.yaml
@@ -35,4 +35,4 @@ jobs:
         with:
           context: .
           push: true
-          tags: hal9ai/hal9-runtime:latest,hal9ai/hal9-runtime:0.0.1
+          tags: hal9ai/hal9-runtime:latest,hal9ai/hal9-runtime:0.0.2
diff --git a/requirements.txt b/requirements.txt
index 70c8379..06706e1 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -17,3 +17,5 @@ httpx
 imageio[opencv]
 imageio[pyav]
 Pillow
+pytorch-lightning
+