Update basic_training.mdx (#2639)

Add 'import os'
This commit is contained in:
M. Tolga Cangöz 2023-03-10 16:19:12 +03:00 committed by GitHub
parent 9360bb94c3
commit f1ab955f64
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 1 deletions

View File

@ -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):
@ -411,4 +412,4 @@ Unconditional image generation is one example of a task that can be trained. You
* [Textual Inversion](./training/text_inversion), an algorithm that teaches a model a specific visual concept and integrates it into the generated image.
* [DreamBooth](./training/dreambooth), a technique for generating personalized images of a subject given several input images of the subject.
* [Guide](./training/text2image) to finetuning a Stable Diffusion model on your own dataset.
* [Guide](./training/lora) to using LoRA, a memory-efficient technique for finetuning really large models faster.
* [Guide](./training/lora) to using LoRA, a memory-efficient technique for finetuning really large models faster.