alexkueck commited on
Commit
e2103dc
1 Parent(s): 9820909

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -15,9 +15,11 @@ from transformers import AutoTokenizer, AutoModelForCausalLM, pipeline
15
 
16
  #Alternativ mit beliebigen Modellen:
17
  #base_model = "project-baize/baize-v2-7b"
18
- base_model = "HuggingFaceH4/starchat-alpha"
19
  tokenizer,model,device = load_tokenizer_and_model(base_model)
20
 
 
 
21
 
22
  ########################################################################
23
  #Chat KI nutzen, um Text zu generieren...
 
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)
22
+ #print( pipe("def hello():") )
23
 
24
  ########################################################################
25
  #Chat KI nutzen, um Text zu generieren...