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__"
|
| 154 |
})
|
| 155 |
|
| 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__"]
|
| 154 |
})
|
| 155 |
|
| 156 |
add_multiview_processor(
|