Update oms_module/unet_2d_condition_woct.py

oms_module/unet_2d_condition_woct.py

@@ -159,6 +159,7 @@ class UNet2DConditionWoCTModel(ModelMixin, ConfigMixin, UNet2DConditionLoadersMi
         conv_out_kernel: int = 3,
         mid_block_only_cross_attention: Optional[bool] = None,
         cross_attention_norm: Optional[str] = None,
+        subfolder: Optional[str] = None,
     ):
         super().__init__()