Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -4,7 +4,7 @@ import torch
|
|
4 |
from transformers import AutoModel, AutoTokenizer, StoppingCriteria, StoppingCriteriaList, TextIteratorStreamer
|
5 |
from threading import Thread
|
6 |
|
7 |
-
MODEL_PATH = os.environ.get('MODEL_PATH', "
|
8 |
TOKENIZER_PATH = os.environ.get("TOKENIZER_PATH", MODEL_PATH)
|
9 |
|
10 |
tokenizer = AutoTokenizer.from_pretrained(TOKENIZER_PATH, trust_remote_code=True)
|
|
|
4 |
from transformers import AutoModel, AutoTokenizer, StoppingCriteria, StoppingCriteriaList, TextIteratorStreamer
|
5 |
from threading import Thread
|
6 |
|
7 |
+
MODEL_PATH = os.environ.get('MODEL_PATH', "ClueAI/ChatYuan-large-v2")
|
8 |
TOKENIZER_PATH = os.environ.get("TOKENIZER_PATH", MODEL_PATH)
|
9 |
|
10 |
tokenizer = AutoTokenizer.from_pretrained(TOKENIZER_PATH, trust_remote_code=True)
|