huanngzh commited on
Commit
cff651e
1 Parent(s): f38509e

fix confusing warning

Browse files
mvadapter/pipelines/pipeline_mvadapter_i2mv_sdxl.py CHANGED
@@ -906,6 +906,14 @@ class MVAdapterI2MVSDXLPipeline(StableDiffusionXLPipeline, CustomAdapterMixin):
906
  use_mv=True,
907
  use_ref=True,
908
  ),
 
 
 
 
 
 
 
 
909
  )
910
 
911
  # copy decoupled attention weights from original unet
 
906
  use_mv=True,
907
  use_ref=True,
908
  ),
909
+ set_cross_attn_proc_func=lambda name, hs, cad, ap: self_attn_processor(
910
+ query_dim=hs,
911
+ inner_dim=hs,
912
+ num_views=num_views,
913
+ name=name,
914
+ use_mv=False,
915
+ use_ref=False,
916
+ ),
917
  )
918
 
919
  # copy decoupled attention weights from original unet
mvadapter/pipelines/pipeline_mvadapter_t2mv_sdxl.py CHANGED
@@ -747,6 +747,14 @@ class MVAdapterT2MVSDXLPipeline(StableDiffusionXLPipeline, CustomAdapterMixin):
747
  use_mv=True,
748
  use_ref=False,
749
  ),
 
 
 
 
 
 
 
 
750
  )
751
 
752
  # copy decoupled attention weights from original unet
 
747
  use_mv=True,
748
  use_ref=False,
749
  ),
750
+ set_cross_attn_proc_func=lambda name, hs, cad, ap: self_attn_processor(
751
+ query_dim=hs,
752
+ inner_dim=hs,
753
+ num_views=num_views,
754
+ name=name,
755
+ use_mv=False,
756
+ use_ref=False,
757
+ ),
758
  )
759
 
760
  # copy decoupled attention weights from original unet