hugo1234 commited on
Commit
fc5fac6
1 Parent(s): e5100d9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -16,8 +16,10 @@ from presets import *
16
  #chatbot = hugchat.ChatBot(cookie_path="cookies.json")
17
 
18
  #base_model = "project-baize/baize-v2-7b"
19
- base_model = "EleutherAI/gpt-neo-1.3B"
20
  #base_model = "EleutherAI/gpt-neo-2.7B"
 
 
21
  tokenizer,model,device = load_tokenizer_and_model(base_model)
22
 
23
 
 
16
  #chatbot = hugchat.ChatBot(cookie_path="cookies.json")
17
 
18
  #base_model = "project-baize/baize-v2-7b"
19
+ #base_model = "EleutherAI/gpt-neo-1.3B"
20
  #base_model = "EleutherAI/gpt-neo-2.7B"
21
+ base_model = "dbmdz/electra-base-italian-xxl-cased-discriminator"
22
+
23
  tokenizer,model,device = load_tokenizer_and_model(base_model)
24
 
25