alexkueck commited on
Commit
a45a332
1 Parent(s): b6c864f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -15,8 +15,7 @@ from transformers import AutoTokenizer, AutoModelForCausalLM, pipeline
15
 
16
  #Alternativ mit beliebigen Modellen:
17
  #base_model = "project-baize/baize-v2-7b"
18
- #base_model = "bigcode/starcoder"
19
- base_model = "TheBloke/airoboros-13B-HF"
20
  tokenizer,model,device = load_tokenizer_and_model(base_model)
21
 
22
  #pipe = pipeline("text-generation", model=model, tokenizer=tokenizer, device=0)
 
15
 
16
  #Alternativ mit beliebigen Modellen:
17
  #base_model = "project-baize/baize-v2-7b"
18
+ base_model = "bigcode/starcoder"
 
19
  tokenizer,model,device = load_tokenizer_and_model(base_model)
20
 
21
  #pipe = pipeline("text-generation", model=model, tokenizer=tokenizer, device=0)