Johnyquest7 commited on
Commit
c0660d2
1 Parent(s): 3e8e7f4

Update app.py

Browse files

internlm/internlm2_5-7b-chat

Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -5,7 +5,7 @@ import spaces
5
  from transformers import AutoModelForCausalLM, AutoTokenizer, TextIteratorStreamer
6
 
7
 
8
- TITLE = "<h1><center>Chat with microsoft/Phi-3-mini-4k-instructt</center></h1>"
9
 
10
  DESCRIPTION = "<h3><center>An experimental space.</center></h3>"
11
 
@@ -102,8 +102,8 @@ CSS = """
102
  """
103
 
104
 
105
- tokenizer = AutoTokenizer.from_pretrained("microsoft/Phi-3-mini-4k-instruct")
106
- model = AutoModelForCausalLM.from_pretrained("microsoft/Phi-3-mini-4k-instruct", torch_dtype="auto", device_map="auto",trust_remote_code=True)
107
 
108
 
109
  @spaces.GPU
 
5
  from transformers import AutoModelForCausalLM, AutoTokenizer, TextIteratorStreamer
6
 
7
 
8
+ TITLE = "<h1><center>Chat with internlm/internlm2_5-7b-chat</center></h1>"
9
 
10
  DESCRIPTION = "<h3><center>An experimental space.</center></h3>"
11
 
 
102
  """
103
 
104
 
105
+ tokenizer = AutoTokenizer.from_pretrained("internlm/internlm2_5-7b-chat")
106
+ model = AutoModelForCausalLM.from_pretrained("internlm/internlm2_5-7b-chat", torch_dtype="auto", device_map="auto",trust_remote_code=True)
107
 
108
 
109
  @spaces.GPU