blanchon
/

dc_flux_krea_diffusers

image-generation

Model card Files Files and versions

blanchon commited on Jan 4

Commit

fa6e2a0

·

verified ·

1 Parent(s): 51b12c5

Update pipeline.py

Files changed (1) hide show

pipeline.py +1 -1

pipeline.py CHANGED Viewed

@@ -42,7 +42,7 @@ from diffusers.utils import (
 from diffusers.utils.torch_utils import randn_tensor
 from diffusers.pipelines.pipeline_utils import DiffusionPipeline
 from diffusers.pipelines.flux.pipeline_output import FluxPipelineOutput
-from fluxdcgen_transformer import DCFluxTransformer2DModel
 if is_torch_xla_available():
     import torch_xla.core.xla_model as xm

 from diffusers.utils.torch_utils import randn_tensor
 from diffusers.pipelines.pipeline_utils import DiffusionPipeline
 from diffusers.pipelines.flux.pipeline_output import FluxPipelineOutput
+from .fluxdcgen_transformer import DCFluxTransformer2DModel
 if is_torch_xla_available():
     import torch_xla.core.xla_model as xm