Reverent commited on
Commit
b60344d
·
verified ·
1 Parent(s): 3abfbb6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -19,7 +19,7 @@ def warmer():
19
  )
20
  end_time = time.time()
21
  duration = end_time - start_time
22
- print("Time elapsed for Pygmalion model : ", duration)
23
  if duration > 90: # Check if duration to generate tokens exceeds 1 minutes
24
  restart()
25
 
@@ -31,7 +31,7 @@ def warmer():
31
  )
32
  end_time = time.time()
33
  duration = end_time - start_time
34
- print("Time elapsed for WizardLM model : ", duration)
35
  if duration > 90: # Check if duration to generate tokens exceeds 1.4 minutes
36
  restart()
37
 
 
19
  )
20
  end_time = time.time()
21
  duration = end_time - start_time
22
+ print("Time elapsed for Mistral model : ", duration)
23
  if duration > 90: # Check if duration to generate tokens exceeds 1 minutes
24
  restart()
25
 
 
31
  )
32
  end_time = time.time()
33
  duration = end_time - start_time
34
+ print("Time elapsed for Starling model : ", duration)
35
  if duration > 90: # Check if duration to generate tokens exceeds 1.4 minutes
36
  restart()
37