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...
|