baalajimaestro/llama.cpp
1
0
Fork 0
Code Issues Pull requests Packages Projects Releases Wiki Activity

Update variable name

Browse source
This commit is contained in:
Andrei Betlen 2023-05-16 18:07:25 -04:00
parent a3352923c7
commit a7c9e38287
1 changed files with 6 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

12
llama_cpp/llama.py
View file

@ -532,7 +532,7 @@ class Llama:
An embedding object.
"""
assert self.ctx is not None
_model: str = model if model is not None else self.model_path
model_name: str = model if model is not None else self.model_path
if self.params.embedding == False:
raise RuntimeError(
@ -562,7 +562,7 @@ class Llama:
"index": 0,
}
],
"model": _model,
"model": model_name,
"usage": {
"prompt_tokens": n_tokens,
"total_tokens": n_tokens,
@ -612,7 +612,7 @@ class Llama:
text: bytes = b""
returned_characters: int = 0
stop = stop if stop is not None else []
_model: str = model if model is not None else self.model_path
model_name: str = model if model is not None else self.model_path
if self.verbose:
llama_cpp.llama_reset_timings(self.ctx)
@ -711,7 +711,7 @@ class Llama:
"id": completion_id,
"object": "text_completion",
"created": created,
"model": _model,
"model": model_name,
"choices": [
{
"text": text[start:].decode("utf-8", errors="ignore"),
@ -740,7 +740,7 @@ class Llama:
"id": completion_id,
"object": "text_completion",
"created": created,
"model": _model,
"model": model_name,
"choices": [
{
"text": text[returned_characters:].decode(
@ -810,7 +810,7 @@ class Llama:
"id": completion_id,
"object": "text_completion",
"created": created,
"model": _model,
"model": model_name,
"choices": [
{
"text": text_str,