diff --git a/requirements.txt b/requirements.txt
index 49539e2..f0a66ae 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -8,4 +8,4 @@ Requests~=2.32
 transformers~=4.52
 utils==1.0.*
 hf_transfer>=0.1.9
-git+https://github.com/vast-ai/vast-sdk.git@remove-input
\ No newline at end of file
+git+https://github.com/vast-ai/vast-sdk.git@worker-sdk
\ No newline at end of file