kz209 commited on
Commit
52ac7d2
·
1 Parent(s): 93ba179

modified: pages/summarization_example.py

Browse files
Files changed (1) hide show
  1. pages/summarization_example.py +1 -1
pages/summarization_example.py CHANGED
@@ -35,7 +35,7 @@ summarization: """ # noqa: E501
35
  messages = [
36
  {"role": "user", "content": content},
37
  ]
38
- pipe = pipeline("text-generation", model="microsoft/Phi-3-mini-4k-instruct", trust_remote_code=True)
39
  answer = pipe(messages)
40
 
41
  # answer = lm.chat.completions.create(
 
35
  messages = [
36
  {"role": "user", "content": content},
37
  ]
38
+ pipe = pipeline("text-generation", model="microsoft/Phi-3-mini-4k-instruct", trust_remote_code=True, max_length=500)
39
  answer = pipe(messages)
40
 
41
  # answer = lm.chat.completions.create(