Spaces:
Runtime error
Runtime error
Refactor to use upstream automodel.
Browse files
app.py
CHANGED
@@ -19,7 +19,7 @@ T4级别的GPU已经可以很胜任这个任务了。
|
|
19 |
import torch
|
20 |
import sys
|
21 |
|
22 |
-
from transformers import AutoTokenizer, GenerationConfig,
|
23 |
|
24 |
model = AutoModel.from_pretrained("THUDM/chatglm-6b", trust_remote_code=True).float()
|
25 |
tokenizer = AutoTokenizer.from_pretrained("THUDM/chatglm-6b", trust_remote_code=True)
|
|
|
19 |
import torch
|
20 |
import sys
|
21 |
|
22 |
+
from transformers import AutoTokenizer, GenerationConfig, AutoModel
|
23 |
|
24 |
model = AutoModel.from_pretrained("THUDM/chatglm-6b", trust_remote_code=True).float()
|
25 |
tokenizer = AutoTokenizer.from_pretrained("THUDM/chatglm-6b", trust_remote_code=True)
|