Update src/pipeline.py
Browse files- src/pipeline.py +1 -1
src/pipeline.py
CHANGED
|
@@ -1,4 +1,4 @@
|
|
| 1 |
-
|
| 2 |
from diffusers.image_processor import VaeImageProcessor
|
| 3 |
from diffusers.schedulers import FlowMatchEulerDiscreteScheduler
|
| 4 |
from huggingface_hub.constants import HF_HUB_CACHE
|
|
|
|
| 1 |
+
from diffusers import FluxPipeline, AutoencoderKL, AutoencoderTiny
|
| 2 |
from diffusers.image_processor import VaeImageProcessor
|
| 3 |
from diffusers.schedulers import FlowMatchEulerDiscreteScheduler
|
| 4 |
from huggingface_hub.constants import HF_HUB_CACHE
|