Skip to content

Commit a18dea6

Browse files
committed
fix(sglang): fixed the backend name in the log
1 parent ef1a86c commit a18dea6

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

mellea/backends/sglang.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -204,7 +204,7 @@ def _generate_from_context_standard(
204204
assert isinstance(generate_logs, list)
205205
generate_log = GenerateLog()
206206
generate_log.prompt = ctx_as_chat
207-
generate_log.backend = f"hf::{self.model_id!s}"
207+
generate_log.backend = f"sglang::{self.model_id!s}"
208208
generate_log.model_options = model_options
209209
generate_log.date = datetime.datetime.now()
210210
generate_log.model_output = decoded_result
@@ -256,7 +256,7 @@ def _generate_from_raw(
256256
for i in range(len(prompts)):
257257
generate_log = GenerateLog()
258258
generate_log.prompt = prompts[i]
259-
generate_log.backend = f"hf::{self.model_id!s}"
259+
generate_log.backend = f"sglang::{self.model_id!s}"
260260
generate_log.model_options = model_options
261261
generate_log.date = date
262262
generate_log.model_output = decoded_results

0 commit comments

Comments
 (0)