alexkueck commited on
Commit
e2ca445
·
1 Parent(s): 6679480

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -126,7 +126,7 @@ base_model = "EleutherAI/gpt-neo-1.3B" #load_8bit = False (in load_tokenizer_
126
  # Load model directly
127
  #####################################################
128
  #Tokenizer und Model laden
129
- tokenizer,model,device = load_tokenizer_and_model(base_model, True)
130
  #tokenizer.add_special_tokens({'pad_token': '[PAD]'}) #not necessary with fast Toekenizers like GPT2
131
  '''
132
  #für Baize.... da spezieller Tokenizer geladen werden muss...
 
126
  # Load model directly
127
  #####################################################
128
  #Tokenizer und Model laden
129
+ tokenizer,model,device = load_tokenizer_and_model(base_model, False)
130
  #tokenizer.add_special_tokens({'pad_token': '[PAD]'}) #not necessary with fast Toekenizers like GPT2
131
  '''
132
  #für Baize.... da spezieller Tokenizer geladen werden muss...