abetlen / abetlen/llama-cpp-python
streaming not working
- 主要語言
- Python
- 星號
- 10.6k
- 分支
- 1.4k
- PR 合併指標
- PR 指標待擷取
描述
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
貢獻指南
評估
這個 Issue 還沒有評估資料。