abenkbp commited on
Commit
f084a4e
·
1 Parent(s): 05c511f
Files changed (1) hide show
  1. data/models/llama3-1-8b.py +1 -1
data/models/llama3-1-8b.py CHANGED
@@ -22,7 +22,7 @@ pipeline = transformers.pipeline(
22
  )
23
 
24
  @app.route('/chat', methods=['POST'])
25
- @spaces.GPU()
26
  def chat_completion():
27
  data = request.json
28
 
 
22
  )
23
 
24
  @app.route('/chat', methods=['POST'])
25
+ @spaces.GPU(duration=60)
26
  def chat_completion():
27
  data = request.json
28