hsuwill000 commited on
Commit
39aecff
·
verified ·
1 Parent(s): 8886195

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -0
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