Spaces:
Running
on
Zero
Running
on
Zero
AlekseyCalvin
commited on
Commit
•
b3b6a55
1
Parent(s):
5164408
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
|