Yash Sachdeva commited on
Commit
da45991
1 Parent(s): c31fdd4

accelerate

Browse files
Files changed (1) hide show
  1. Dockerfile +2 -0
Dockerfile CHANGED
@@ -14,6 +14,8 @@ RUN pip install torch
14
 
15
  #Install pytorch ends
16
 
 
 
17
  # Install requirements.txt
18
  RUN pip install --no-cache-dir --upgrade -r /requirements.txt
19
 
 
14
 
15
  #Install pytorch ends
16
 
17
+ RUN pip install accelerate
18
+
19
  # Install requirements.txt
20
  RUN pip install --no-cache-dir --upgrade -r /requirements.txt
21