Update app.py
Browse files
app.py
CHANGED
@@ -95,8 +95,10 @@ trainer = Trainer(
|
|
95 |
tokenizer=tokenizer,
|
96 |
)
|
97 |
|
98 |
-
|
99 |
-
trainer.train()
|
|
|
|
|
100 |
|
101 |
model.save_pretrained("./fine_tuned_llama2")
|
102 |
tokenizer.save_pretrained("./fine_tuned_llama2")
|
|
|
95 |
tokenizer=tokenizer,
|
96 |
)
|
97 |
|
98 |
+
def perform_training():
|
99 |
+
trainer.train()
|
100 |
+
|
101 |
+
perform_training()
|
102 |
|
103 |
model.save_pretrained("./fine_tuned_llama2")
|
104 |
tokenizer.save_pretrained("./fine_tuned_llama2")
|