jixin0101 commited on
Commit
f66316a
·
verified ·
1 Parent(s): 6fa3be1

Update pipeline_objectclear.py

Browse files
Files changed (1) hide show
  1. 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