tomato commited on
Commit
76b83e6
1 Parent(s): bdb909e

remove [max_new_tokens]

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -5,7 +5,7 @@ from transformers import AutoTokenizer, AutoModel
5
  MODEL_NAME = "silver/chatglm-6b-slim"
6
 
7
  tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME, trust_remote_code=True)
8
- model = AutoModel.from_pretrained(MODEL_NAME, max_new_tokens=8192, trust_remote_code=True).float()
9
 
10
  def summarize(transcript, sentence_count):
11
  history = []
 
5
  MODEL_NAME = "silver/chatglm-6b-slim"
6
 
7
  tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME, trust_remote_code=True)
8
+ model = AutoModel.from_pretrained(MODEL_NAME, trust_remote_code=True).float()
9
 
10
  def summarize(transcript, sentence_count):
11
  history = []