diff --git a/examples/streaming-falcon-7B.mdx b/examples/streaming-falcon-7B.mdx
index 42ae31c1..3e50c09e 100644
--- a/examples/streaming-falcon-7B.mdx
+++ b/examples/streaming-falcon-7B.mdx
@@ -29,7 +29,7 @@ transformers = "git+https://github.com/huggingface/transformers.git"
 accelerate = "git+https://github.com/huggingface/accelerate.git"
 bitsandbytes = ""
 sentencepiece = ""
-torch = ""
+torch = "2.1.0"
 ```
 
 Our **main.py** file will contain our main Python code. This is a relatively simple implementation, so we can do everything in 1 file. We would like a user to send in a link to a YouTube video with a question and return to them the answer as well as the time segment of where we got that response.
@@ -153,7 +153,7 @@ transformers = "git+https://github.com/huggingface/transformers.git"
 accelerate = "git+https://github.com/huggingface/accelerate.git"
 bitsandbytes = ""
 sentencepiece = ""
-torch = ""
+torch = "2.1.0"
 
 [cerebrium.requirements.conda]