alexkueck commited on
Commit
e9d94bd
1 Parent(s): 050d909

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -5
app.py CHANGED
@@ -16,11 +16,11 @@ from presets import *
16
  #chatbot = hugchat.ChatBot(cookie_path="cookies.json")
17
 
18
  #Alternativ mit beliebigen Modellen:
19
- #base_model = "project-baize/baize-v2-7b"
20
- #base_model = "TheBloke/airoboros-13B-HF"
21
- base_model = "EleutherAI/gpt-neo-1.3B"
22
- #base_model = "TheBloke/airoboros-13B-HF"
23
- tokenizer,model,device = load_tokenizer_and_model(base_model)
24
 
25
 
26
  ########################################################################
 
16
  #chatbot = hugchat.ChatBot(cookie_path="cookies.json")
17
 
18
  #Alternativ mit beliebigen Modellen:
19
+ #base_model = "project-baize/baize-v2-7b" #load_8bit = False (in load_tokenizer_and_model)
20
+ #base_model = "TheBloke/airoboros-13B-HF" #load_8bit = False (in load_tokenizer_and_model)
21
+ #base_model = "EleutherAI/gpt-neo-1.3B" #load_8bit = False (in load_tokenizer_and_model)
22
+ base_model = "TheBloke/airoboros-13B-HF" #load_8bit = True
23
+ tokenizer,model,device = load_tokenizer_and_model(base_model,True)
24
 
25
 
26
  ########################################################################