DDPM training example

pipeline_glide.py

@@ -24,6 +24,9 @@ import torch.utils.checkpoint
 from torch import nn
 
 import tqdm
+from diffusers.models import GLIDESuperResUNetModel, GLIDETextToImageUNetModel
+from diffusers.pipeline_utils import DiffusionPipeline
+from diffusers.schedulers import ClassifierFreeGuidanceScheduler, DDIMScheduler
 from transformers import CLIPConfig, CLIPModel, CLIPTextConfig, CLIPVisionConfig, GPT2Tokenizer
 from transformers.activations import ACT2FN
 from transformers.modeling_outputs import BaseModelOutput, BaseModelOutputWithPooling
@@ -36,10 +39,6 @@ from transformers.utils import (
     replace_return_docstrings,
 )
 
-from diffusers.models import GLIDESuperResUNetModel, GLIDETextToImageUNetModel
-from diffusers.pipeline_utils import DiffusionPipeline
-from diffusers.schedulers import ClassifierFreeGuidanceScheduler, DDIMScheduler
-
 
 #####################
 # START OF THE CLIP MODEL COPY-PASTE (with a modified attention module)