Spaces:
Paused
Paused
tomato
commited on
Commit
·
76b83e6
1
Parent(s):
bdb909e
remove [max_new_tokens]
Browse files
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,
|
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 = []
|