Update src/pipeline.py
Browse files- src/pipeline.py +1 -1
src/pipeline.py
CHANGED
|
@@ -56,7 +56,7 @@ from diffusers.utils.torch_utils import randn_tensor
|
|
| 56 |
from diffusers.pipelines.pipeline_utils import DiffusionPipeline
|
| 57 |
from diffusers.pipelines.flux.pipeline_output import FluxPipelineOutput
|
| 58 |
|
| 59 |
-
|
| 60 |
# preconfigs
|
| 61 |
import os
|
| 62 |
os.environ['PYTORCH_CUDA_ALLOC_CONF']="expandable_segments:True"
|
|
|
|
| 56 |
from diffusers.pipelines.pipeline_utils import DiffusionPipeline
|
| 57 |
from diffusers.pipelines.flux.pipeline_output import FluxPipelineOutput
|
| 58 |
|
| 59 |
+
import torch.utils.benchmark as benchmark
|
| 60 |
# preconfigs
|
| 61 |
import os
|
| 62 |
os.environ['PYTORCH_CUDA_ALLOC_CONF']="expandable_segments:True"
|