Spaces:
Running on Zero
Running on Zero
stamp raw responses as oscar-1
Browse files- 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":
|
|
|
|
| 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}}
|