AlekseyCalvin commited on
Commit
3f2133d
1 Parent(s): 49b4dc1

Update custom_pipeline.py

Browse files
Files changed (1) hide show
  1. custom_pipeline.py +1 -1
custom_pipeline.py CHANGED
@@ -1,7 +1,7 @@
1
  import torch
2
  import numpy as np
3
  from diffusers import FlowMatchEulerDiscreteScheduler
4
- from diffusers.pipelines.flux.pipeline_output import FluxPipeline, FluxPipelineOutput
5
  from typing import Any, Callable, Dict, List, Optional, Union
6
  from PIL import Image
7
  from diffusers.pipelines.flux.pipeline_flux import calculate_shift, retrieve_timesteps
 
1
  import torch
2
  import numpy as np
3
  from diffusers import FlowMatchEulerDiscreteScheduler
4
+ from diffusers.pipelines.flux.pipeline_output import FluxPipelineOutput
5
  from typing import Any, Callable, Dict, List, Optional, Union
6
  from PIL import Image
7
  from diffusers.pipelines.flux.pipeline_flux import calculate_shift, retrieve_timesteps