x
This commit is contained in:
+1
-1
@@ -39,7 +39,7 @@ class InferenceEngine:
|
||||
return GenerateResponse(
|
||||
text=completion,
|
||||
prompt=req.prompt,
|
||||
model=self.settings.model_name,
|
||||
model=self.settings.served_model_name or self.settings.model_name,
|
||||
usage_prompt_tokens=usage_prompt,
|
||||
usage_completion_tokens=usage_completion,
|
||||
usage_total_tokens=usage_prompt + usage_completion,
|
||||
|
||||
Reference in New Issue
Block a user