fix params save bug
Browse files- 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(
|