mgoeckel commited on
Commit
00826ae
路
verified 路
1 Parent(s): 9a59c28

stamp raw responses as oscar-1

Browse files
Files changed (1) hide show
  1. rl_agent_api.py +4 -1
rl_agent_api.py CHANGED
@@ -6,6 +6,9 @@ import os
6
  import numpy as np
7
  import torch
8
 
 
 
 
9
  from rl_common import (QTYPES, amp_dtype, build_model, build_sequence, collate_items, confidence_from_probs,
10
  render_options, temp_bucket)
11
 
@@ -127,4 +130,4 @@ class RLAgent:
127
  "confidence": round(confidence_from_probs(p, k), 4), "rl_agent": ext}
128
  else:
129
  answers[qid] = {"type": "noul", "noul": round(float(p[1]), 4), "rl_agent": ext}
130
- return {"model": "laya", "answers": answers, "usage": {"input_tokens": n_tokens, "output_tokens": 0}}
 
6
  import numpy as np
7
  import torch
8
 
9
+ """Checkpoint family name stamped into raw responses."""
10
+ MODEL_NAME = "oscar-1"
11
+
12
  from rl_common import (QTYPES, amp_dtype, build_model, build_sequence, collate_items, confidence_from_probs,
13
  render_options, temp_bucket)
14
 
 
130
  "confidence": round(confidence_from_probs(p, k), 4), "rl_agent": ext}
131
  else:
132
  answers[qid] = {"type": "noul", "noul": round(float(p[1]), 4), "rl_agent": ext}
133
+ return {"model": MODEL_NAME, "answers": answers, "usage": {"input_tokens": n_tokens, "output_tokens": 0}}