zencorn commited on
Commit
9ff78e1
·
1 Parent(s): 6265b0a

Update application file

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -32,7 +32,7 @@ from transformers import AutoTokenizer, AutoModelForCausalLM # isort: skip
32
  logger = logging.get_logger(__name__)
33
  '''model_name_or_path="/root/finetune/work_dirs/assistTuner/merged"'''
34
 
35
- model_name_or_path = "/root/finetune/work_dirs/assistTuner/merged"
36
  @dataclass
37
  class GenerationConfig:
38
  # this config is used for chat to provide more diversity
 
32
  logger = logging.get_logger(__name__)
33
  '''model_name_or_path="/root/finetune/work_dirs/assistTuner/merged"'''
34
 
35
+ model_name_or_path = "Jianfei217/XtuneModel"
36
  @dataclass
37
  class GenerationConfig:
38
  # this config is used for chat to provide more diversity