diff --git a/02.forge.sh b/02.forge.sh
index 43578bc..1e51274 100644
--- a/02.forge.sh
+++ b/02.forge.sh
@@ -53,11 +53,13 @@ if [ ! -f "$SD02_DIR/parameters.forge.txt" ]; then
 fi
 
 pip install --upgrade pip
+pip install coloredlogs flatbuffers numpy packaging protobuf==3.20.3 sympy
 pip install packaging
 pip install onnxruntime-gpu --extra-index-url https://aiinfra.pkgs.visualstudio.com/PublicPackages/_packaging/onnxruntime-cuda-12/pypi/simple/
-pip install insightface 
+pip install insightface
 pip install basicsr
-#pip install protobuf==3.20.3
+pip install torch==2.2.0 torchvision==0.17.0 torchaudio==2.2.0 --index-url https://download.pytorch.org/whl/cu121
+pip install xformers --index-url https://download.pytorch.org/whl/cu121
 
 # Merge Models, vae, lora, and hypernetworks, and outputs
 # Ignore move errors if they occur
diff --git a/02.sh b/02.sh
index a0d1402..35e1305 100755
--- a/02.sh
+++ b/02.sh
@@ -53,11 +53,14 @@ if [ ! -f "$SD02_DIR/parameters.txt" ]; then
 fi
 
 pip install --upgrade pip
+pip install coloredlogs flatbuffers numpy packaging protobuf==3.20.3 sympy
 pip install packaging
 pip install onnxruntime-gpu --extra-index-url https://aiinfra.pkgs.visualstudio.com/PublicPackages/_packaging/onnxruntime-cuda-12/pypi/simple/
 pip install insightface
-pip install basicsr 
-#pip install protobuf==3.20.3
+pip install basicsr
+pip install torch==2.2.0 torchvision==0.17.0 torchaudio==2.2.0 --index-url https://download.pytorch.org/whl/cu121
+pip install xformers --index-url https://download.pytorch.org/whl/cu121
+
 
 # Merge Models, vae, lora, and hypernetworks, and outputs
 # Ignore move errors if they occur
diff --git a/Dockerfile b/Dockerfile
index 4ddbc3a..d2b7f99 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -34,7 +34,7 @@ RUN apt-get update -y -q=2 && \
     build-essential \
     python3-opencv \
     libopencv-dev \
-    dotnet-sdk-7.0 \
+    dotnet-sdk-8.0 \
     git && \
     apt-get clean && \
     rm -rf /var/lib/apt/lists/* /tmp/* /var/tmp/*