jonACE commited on
Commit
66ee08b
·
verified ·
1 Parent(s): ac97ba7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -2
app.py CHANGED
@@ -95,8 +95,10 @@ trainer = Trainer(
95
  tokenizer=tokenizer,
96
  )
97
 
98
- @unsloth.optimize
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")