Spaces:
Paused
Paused
Update app.py
Browse files
app.py
CHANGED
@@ -137,7 +137,7 @@ config = LoraConfig(
|
|
137 |
bias="none",
|
138 |
task_type="CAUSAL_LM",
|
139 |
)
|
140 |
-
|
141 |
model = get_peft_model(model, config)
|
142 |
#da schneller Tokenizer-> Attention Mask setzen
|
143 |
tokenizer.pad_token_id = 0
|
|
|
137 |
bias="none",
|
138 |
task_type="CAUSAL_LM",
|
139 |
)
|
140 |
+
config.save_pretrained(OUTPUT_DIR)
|
141 |
model = get_peft_model(model, config)
|
142 |
#da schneller Tokenizer-> Attention Mask setzen
|
143 |
tokenizer.pad_token_id = 0
|