Luffuly
/

unique3d-mvimage-diffuser

StableDiffusionImage2MVCustomPipeline

Model card Files Files and versions

Luffuly commited on Jan 2

Commit

204dc62

·

1 Parent(s): cc10fb7

fix params save bug

Files changed (1) hide show

unet/mv_unet.py +1 -1

unet/mv_unet.py CHANGED Viewed

@@ -150,7 +150,7 @@ class UnifieldWrappedUNet(UNet2DConditionModel):
         ):
         super().__init__(**{
                 k: v for k, v in locals().items() if k not in
-                ["self", "kwargs", "__class__", "n_views", "num_modalities", "latent_size", "multiview_chain_pose"]
             })
         self.n_views = n_views
         add_multiview_processor(

         ):
         super().__init__(**{
                 k: v for k, v in locals().items() if k not in
+                ["self", "kwargs", "__class__", "n_views", "num_modalities", "latent_size", "multiview_chain_pose", "multiview_attn_position"]
             })
         self.n_views = n_views
         add_multiview_processor(