Spaces:
Running
on
Zero
Running
on
Zero
Update pipeline_objectclear.py
Browse files- pipeline_objectclear.py +1 -0
pipeline_objectclear.py
CHANGED
@@ -56,6 +56,7 @@ from diffusers.utils import (
|
|
56 |
from diffusers.utils.torch_utils import randn_tensor
|
57 |
from diffusers.pipelines.pipeline_utils import DiffusionPipeline, StableDiffusionMixin
|
58 |
from diffusers.pipelines.stable_diffusion_xl.pipeline_output import StableDiffusionXLPipelineOutput
|
|
|
59 |
|
60 |
from model import CLIPImageEncoder, PostfuseModule
|
61 |
from utils import attention_guided_fusion
|
|
|
56 |
from diffusers.utils.torch_utils import randn_tensor
|
57 |
from diffusers.pipelines.pipeline_utils import DiffusionPipeline, StableDiffusionMixin
|
58 |
from diffusers.pipelines.stable_diffusion_xl.pipeline_output import StableDiffusionXLPipelineOutput
|
59 |
+
from dataclasses import dataclass
|
60 |
|
61 |
from model import CLIPImageEncoder, PostfuseModule
|
62 |
from utils import attention_guided_fusion
|