Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -18,6 +18,11 @@ pipe = ov_genai.LLMPipeline(model_path, device)
|
|
18 |
tokenizer = pipe.get_tokenizer()
|
19 |
config = ov_genai.GenerationConfig(max_new_tokens=4096)
|
20 |
|
|
|
|
|
|
|
|
|
|
|
21 |
# 推理函式
|
22 |
def generate_response(prompt, model_name):
|
23 |
global pipe, tokenizer
|
|
|
18 |
tokenizer = pipe.get_tokenizer()
|
19 |
config = ov_genai.GenerationConfig(max_new_tokens=4096)
|
20 |
|
21 |
+
def streamer(subword):
|
22 |
+
global output_buffer
|
23 |
+
output_buffer += subword
|
24 |
+
print(subword, end='', flush=True)
|
25 |
+
|
26 |
# 推理函式
|
27 |
def generate_response(prompt, model_name):
|
28 |
global pipe, tokenizer
|