alexkueck commited on
Commit
66627bf
·
1 Parent(s): 58c0d2a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -85,7 +85,9 @@ login(token=os.environ["HF_ACCESS_TOKEN"]) #for read access!!!!
85
  #base_model = "EleutherAI/gpt-neo-1.3B" #load_8bit = False (in load_tokenizer_and_model)
86
  #base_model = "TheBloke/airoboros-13B-HF" #load_8bit = True
87
  #base_model = "TheBloke/vicuna-13B-1.1-HF" #load_8bit = ?
88
- base_model="gpt2-xl"
 
 
89
 
90
  #Tokenizer und Model laden
91
  tokenizer,model,device = load_tokenizer_and_model(base_model, False)
 
85
  #base_model = "EleutherAI/gpt-neo-1.3B" #load_8bit = False (in load_tokenizer_and_model)
86
  #base_model = "TheBloke/airoboros-13B-HF" #load_8bit = True
87
  #base_model = "TheBloke/vicuna-13B-1.1-HF" #load_8bit = ?
88
+ #base_model="gpt2-xl"
89
+ base_model = "stabilityai/stablecode-instruct-alpha-3b" #load_8bit = ?
90
+ # Load model directly
91
 
92
  #Tokenizer und Model laden
93
  tokenizer,model,device = load_tokenizer_and_model(base_model, False)