Fix version of torch

CerebriumAI · Dec 7, 2023 · 21a56fe · 21a56fe
1 parent c4a8948
commit 21a56fe
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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]