Skip to content

Commit 33e054a

Browse files
committed
Update pipeline_bria.py
1 parent 6e8ffd1 commit 33e054a

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

src/diffusers/pipelines/bria/pipeline_bria.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -9,7 +9,8 @@
99

1010
from ...image_processor import VaeImageProcessor
1111
from ...loaders import FluxLoraLoaderMixin
12-
from ...models import AutoencoderKL, BriaTransformer2DModel
12+
from ...models import AutoencoderKL
13+
from ...models import BriaTransformer2DModel
1314
from ...pipelines.flux.pipeline_flux import FluxPipeline, calculate_shift, retrieve_timesteps
1415
from .pipeline_output import BriaPipelineOutput
1516
from ...schedulers import (

0 commit comments

Comments
 (0)