Spaces:
Runtime error
Runtime error
Update pipeline.py
Browse files- pipeline.py +12 -2
pipeline.py
CHANGED
|
@@ -1,7 +1,17 @@
|
|
| 1 |
import torch
|
| 2 |
import numpy as np
|
| 3 |
-
from
|
| 4 |
-
from diffusers import
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 5 |
from diffusers.pipelines.flux.pipeline_output import FluxPipelineOutput
|
| 6 |
from typing import Any, Callable, Dict, List, Optional, Union
|
| 7 |
from PIL import Image
|
|
|
|
| 1 |
import torch
|
| 2 |
import numpy as np
|
| 3 |
+
from transformers import CLIPTextModel, CLIPTokenizer, T5EncoderModel, T5TokenizerFast
|
| 4 |
+
from diffusers import FlowMatchEulerDiscreteScheduler, AutoPipelineForImage2Image, FluxPipeline, FluxTransformer2DModel, VaeImageProcessor
|
| 5 |
+
from diffusers import StableDiffusion3Pipeline, AutoencoderKL, DiffusionPipeline, FluxLoraLoaderMixin, FromSingleFileMixin, SD3LoraLoaderMixin
|
| 6 |
+
from diffusers.utils import (
|
| 7 |
+
USE_PEFT_BACKEND,
|
| 8 |
+
is_torch_xla_available,
|
| 9 |
+
logging,
|
| 10 |
+
replace_example_docstring,
|
| 11 |
+
scale_lora_layers,
|
| 12 |
+
unscale_lora_layers,
|
| 13 |
+
)
|
| 14 |
+
from diffusers.utils.torch_utils import randn_tensor
|
| 15 |
from diffusers.pipelines.flux.pipeline_output import FluxPipelineOutput
|
| 16 |
from typing import Any, Callable, Dict, List, Optional, Union
|
| 17 |
from PIL import Image
|