Update basic_training.mdx (#2639)

Add 'import os'
2023-03-10 16:19:12 +03:00 · 2023-03-10 16:19:12 +03:00 · f1ab955f64
parent 9360bb94c3
commit f1ab955f64
1 changed files with 2 additions and 1 deletions
--- a/docs/source/en/tutorials/basic_training.mdx
+++ b/docs/source/en/tutorials/basic_training.mdx
@ -252,6 +252,7 @@ Then, you'll need a way to evaluate the model. For evaluation, you can use the [
 ```py
 >>> from diffusers import DDPMPipeline
 >>> import math
+>>> import os


 >>> def make_grid(images, rows, cols):