Spaces:

Hunzla
/

fine-tuning

Runtime error

Hunzla commited on Aug 24, 2023

Commit

8a6a0ad

1 Parent(s): 58b6c18

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -202,7 +202,7 @@ trainer.train()
 # Save trained model
 trainer.model.save_pretrained(new_model)
 # Ignore warnings
- logging.set_verbosity(logging.CRITICAL)
 # Run text generation pipeline with our next model
 prompt = "What is a large language model?"

 # Save trained model
 trainer.model.save_pretrained(new_model)
 # Ignore warnings
+logging.set_verbosity(logging.CRITICAL)
 # Run text generation pipeline with our next model
 prompt = "What is a large language model?"