Luffuly commited on
Commit
204dc62
·
1 Parent(s): cc10fb7

fix params save bug

Browse files
Files changed (1) hide show
  1. unet/mv_unet.py +1 -1
unet/mv_unet.py CHANGED
@@ -150,7 +150,7 @@ class UnifieldWrappedUNet(UNet2DConditionModel):
150
  ):
151
  super().__init__(**{
152
  k: v for k, v in locals().items() if k not in
153
- ["self", "kwargs", "__class__", "n_views", "num_modalities", "latent_size", "multiview_chain_pose"]
154
  })
155
  self.n_views = n_views
156
  add_multiview_processor(
 
150
  ):
151
  super().__init__(**{
152
  k: v for k, v in locals().items() if k not in
153
+ ["self", "kwargs", "__class__", "n_views", "num_modalities", "latent_size", "multiview_chain_pose", "multiview_attn_position"]
154
  })
155
  self.n_views = n_views
156
  add_multiview_processor(