Spaces:
Running
Running
tmzh
commited on
Commit
•
113ecf4
1
Parent(s):
6660737
add accelerate package
Browse files- requirements.txt +2 -1
- 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":
|
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",
|