Adding model.config.use_cache = False to prevent warning

utils/notebook_generator.py

@@ -319,6 +319,8 @@ loop and train the model for `{epoch_str}`. The model will be automatically save
 
     code = f"""
 
+model.config.use_cache = False
+
 # start training
 trainer.train()