Spaces:
Runtime error
Runtime error
Commit
•
e9855cf
1
Parent(s):
c5d2851
adding accelerate
Browse files- app.py +1 -1
- requirements.txt +2 -1
app.py
CHANGED
|
@@ -31,4 +31,4 @@ def falcon(input_text):
|
|
| 31 |
return sequences[0]['generated_text']
|
| 32 |
|
| 33 |
iface = gr.Interface(fn=falcon, inputs="text", outputs="text")
|
| 34 |
-
iface.launch()
|
|
|
|
| 31 |
return sequences[0]['generated_text']
|
| 32 |
|
| 33 |
iface = gr.Interface(fn=falcon, inputs="text", outputs="text")
|
| 34 |
+
iface.launch() # To create a public link, set `share=True`
|
requirements.txt
CHANGED
|
@@ -1,3 +1,4 @@
|
|
| 1 |
transformers==4.29.2
|
| 2 |
torch==2.0.1
|
| 3 |
-
einops==0.6.1
|
|
|
|
|
|
| 1 |
transformers==4.29.2
|
| 2 |
torch==2.0.1
|
| 3 |
+
einops==0.6.1
|
| 4 |
+
accelerate==0.19.0
|