taki0112 commited on
Commit
0d4c732
1 Parent(s): fb46c66
Files changed (2) hide show
  1. app.py +1 -1
  2. pipelines/pipeline_controlnet_sd_xl.py +0 -1
app.py CHANGED
@@ -219,7 +219,7 @@ def style_fn(image_path, style_name, content_text, output_number, diffusion_step
219
  use_advanced_sampling = False
220
  use_prompt_as_null = True
221
 
222
- ref_prompt = blip_inf_prompt(origin_real_img)
223
  inf_prompt = inf_object
224
  style_description_neg = None
225
 
 
219
  use_advanced_sampling = False
220
  use_prompt_as_null = True
221
 
222
+ ref_prompt = blip_inf_prompt(origin_real_img).to(device)
223
  inf_prompt = inf_object
224
  style_description_neg = None
225
 
pipelines/pipeline_controlnet_sd_xl.py CHANGED
@@ -48,7 +48,6 @@ if is_invisible_watermark_available():
48
  from diffusers.pipelines.controlnet.multicontrolnet import MultiControlNetModel
49
  import os
50
  from pipelines.inverted_ve_pipeline import CrossFrameAttnProcessor, ACTIVATE_LAYER_CANDIDATE, SharedAttentionProcessor, SharedAttentionProcessor_v2
51
- import gc
52
 
53
  logger = logging.get_logger(__name__) # pylint: disable=invalid-name
54
 
 
48
  from diffusers.pipelines.controlnet.multicontrolnet import MultiControlNetModel
49
  import os
50
  from pipelines.inverted_ve_pipeline import CrossFrameAttnProcessor, ACTIVATE_LAYER_CANDIDATE, SharedAttentionProcessor, SharedAttentionProcessor_v2
 
51
 
52
  logger = logging.get_logger(__name__) # pylint: disable=invalid-name
53