ShinyRO commited on
Commit
2323307
Β·
verified Β·
1 Parent(s): 824d918

test: latency test with lstm model (transformers latency is 450-950ms excluding overhead)

Browse files
Files changed (1) hide show
  1. inference_api.py +1 -1
inference_api.py CHANGED
@@ -35,7 +35,7 @@ logger = logging.getLogger("DKTInferenceEngine")
35
  # Global Configuration
36
  # ──────────────────────────────────────────────────────────────
37
  # Easy Switch Model Choice: 'LSTM' or 'Transformer'
38
- MODEL_TYPE = "Transformer"
39
 
40
  # Thresholds
41
  STRUGGLE_THRESHOLD = 0.50 # Skills below this are flagged as struggling
 
35
  # Global Configuration
36
  # ──────────────────────────────────────────────────────────────
37
  # Easy Switch Model Choice: 'LSTM' or 'Transformer'
38
+ MODEL_TYPE = "LSTM"
39
 
40
  # Thresholds
41
  STRUGGLE_THRESHOLD = 0.50 # Skills below this are flagged as struggling