CH-UP commited on
Commit
753fcce
·
1 Parent(s): 258bb26

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -3,8 +3,8 @@ import gradio as gr
3
  import mdtex2html
4
  from utils import load_model_on_gpus
5
 
6
- tokenizer = AutoTokenizer.from_pretrained("CH-UP/chatglm2-6b-torchkeras-ws", trust_remote_code=True)
7
- model = AutoModel.from_pretrained("CH-UP/chatglm2-6b-torchkeras-ws", trust_remote_code=True).float()
8
  # 多显卡支持,使用下面两行代替上面一行,将num_gpus改为你实际的显卡数量
9
  # from utils import load_model_on_gpus
10
  # model = load_model_on_gpus("THUDM/chatglm2-6b", num_gpus=2)
 
3
  import mdtex2html
4
  from utils import load_model_on_gpus
5
 
6
+ tokenizer = AutoTokenizer.from_pretrained("IPLab/chatglm2-6b-patent-7.30", trust_remote_code=True)
7
+ model = AutoModel.from_pretrained("IPLab/chatglm2-6b-patent-7.30", trust_remote_code=True).float()
8
  # 多显卡支持,使用下面两行代替上面一行,将num_gpus改为你实际的显卡数量
9
  # from utils import load_model_on_gpus
10
  # model = load_model_on_gpus("THUDM/chatglm2-6b", num_gpus=2)