abetlen / abetlen/llama-cpp-python

streaming not working

Offen
#1,604 2 Kommentare 0 Reaktionen 0 zugewiesene Personen Auf GitHub ansehen
Vorherrschende Sprache
Python
Sterne
10.6k
Forks
1.4k
PR-Merge-Kennzahlen
PR-Kennzahlen ausstehend

Beschreibung

Can somebody describe, why streaming is not working? The model provides the answer at once instead of streaming,

llm = Llama(
streaming=True,
model_path="/mistral-7b-chat-int8-v0.2.Q4_K_M.gguf",
n_ctx=32768,
n_threads=4,
n_gpu_layers=20,
callbacks=[StreamingStdOutCallbackHandler()],
verbose=True
)

generation_kwargs = {
"max_tokens": 20000,
"stop": [""],
"streaming": True,
"echo": False,
"top_k": 1
}

llm_response = llm(prompt, **generation_kwargs)

final_result = llm_response["choices"][0]["text"]

return final_result

Beitragsleitfaden

Beitragsleitfaden öffnen

Bewertung

Dieses Issue wurde noch nicht bewertet.

Neue Issues direkt in Ihr Postfach

Eine kurze Übersicht über anfängerfreundliche GitHub-Issues.