diff --git a/.gitignore b/.gitignore
index 6b8bafe..9cd2434 100644
--- a/.gitignore
+++ b/.gitignore
@@ -11,3 +11,4 @@ gui-user.bat
 gui-user.ps1
 .vscode
 wandb
+logs
\ No newline at end of file
diff --git a/setup.bat b/setup.bat
index 9416f83..124d40d 100644
--- a/setup.bat
+++ b/setup.bat
@@ -21,8 +21,7 @@ echo [2] - No (recommanded for most)
 set /p uninstall_choice="Enter your choice (1 or 2): "
 
 if %uninstall_choice%==1 (
-    pip uninstall -y xformers
-    pip uninstall -y torch torchvision
+    pip uninstall -y xformers torch torchvision triton
 )
 
 echo Please choose the version of torch you want to install: