Leon4gr45 commited on
Commit
ab1c9a3
·
verified ·
1 Parent(s): cdc816f

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +2 -1
requirements.txt CHANGED
@@ -15,4 +15,5 @@ pydantic
15
  transformers
16
  accelerate
17
 
18
- # torch is provided by the base Docker image, so it is not needed here
 
 
15
  transformers
16
  accelerate
17
 
18
+ # Add torch back, as the new base image does not include it
19
+ torch==2.3.0