cloneQ commited on
Commit
4838c4e
1 Parent(s): c01dc02

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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-7b-chat-4bit"
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: