tmzh commited on
Commit
113ecf4
1 Parent(s): 6660737

add accelerate package

Browse files
Files changed (2) hide show
  1. requirements.txt +2 -1
  2. tools.py +1 -1
requirements.txt CHANGED
@@ -2,4 +2,5 @@ Flask
2
  requests
3
  transformers
4
  torch
5
- bitsandbytes
 
 
2
  requests
3
  transformers
4
  torch
5
+ bitsandbytes
6
+ accelerate
tools.py CHANGED
@@ -188,7 +188,7 @@ tools = [
188
  "include_adult": {
189
  "type": "boolean",
190
  "description": "Include adult (pornography) content in the results",
191
- "default": false
192
  },
193
  "language": {
194
  "type": "string",
 
188
  "include_adult": {
189
  "type": "boolean",
190
  "description": "Include adult (pornography) content in the results",
191
+ "default": False
192
  },
193
  "language": {
194
  "type": "string",