Spaces:
Paused
Paused
Update app.py
Browse files
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,
|
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...
|