alexkueck commited on
Commit
f1b46b4
·
1 Parent(s): 6a0b199

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -129,13 +129,13 @@ base_model = "EleutherAI/gpt-neo-1.3B" #load_8bit = False (in load_tokenizer_
129
  # Load model directly
130
  #####################################################
131
  #Tokenizer und Model laden
132
- #tokenizer,model,device = load_tokenizer_and_model(base_model, False)
133
 
134
  #################################################
135
  # Few-Shot Training!!!!!!!!!
136
  #################################################
137
  # Load SetFit model from Hub
138
- tokenizer,model,device = load_tokenizer_and_model_setFit(base_model, False)
139
 
140
 
141
  #tokenizer.add_special_tokens({'pad_token': '[PAD]'}) #not necessary with fast Toekenizers like GPT2
 
129
  # Load model directly
130
  #####################################################
131
  #Tokenizer und Model laden
132
+ tokenizer,model,device = load_tokenizer_and_model(base_model, False)
133
 
134
  #################################################
135
  # Few-Shot Training!!!!!!!!!
136
  #################################################
137
  # Load SetFit model from Hub
138
+ #tokenizer,model,device = load_tokenizer_and_model_setFit(base_model, False)
139
 
140
 
141
  #tokenizer.add_special_tokens({'pad_token': '[PAD]'}) #not necessary with fast Toekenizers like GPT2