Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -30,7 +30,7 @@ from transformers.utils import logging
|
|
30 |
from transformers import AutoTokenizer, AutoModelForCausalLM # isort: skip
|
31 |
|
32 |
logger = logging.get_logger(__name__)
|
33 |
-
model_name_or_path="internlm/internlm2_5-
|
34 |
|
35 |
@dataclass
|
36 |
class GenerationConfig:
|
|
|
30 |
from transformers import AutoTokenizer, AutoModelForCausalLM # isort: skip
|
31 |
|
32 |
logger = logging.get_logger(__name__)
|
33 |
+
model_name_or_path="internlm/internlm2_5-1_8b-chat"
|
34 |
|
35 |
@dataclass
|
36 |
class GenerationConfig:
|