Upload TFBilma

modeling_bilma.py

@@ -119,13 +119,13 @@ def bilma(num_enc=6, embed_dim=300, max_length=50, num_heads=6, ff_dim=512, voca
     if include_top:
         fin_output = Dense(vocab_size, use_bias=True, name="bilma/dense_final")(enc_output)
     else:
-        if self.include_head is None:
+        if include_head is None:
             fin_output = enc_output
         else:
             x = enc_output
-            for i, m in enumerate(self.include_head[:-1]):
+            for i, m in enumerate(include_head[:-1]):
                 x = Dense(m, use_bias=True, activation="relu", name=f"bilma/dense_ex_{i}")(x)
-            fin_output = [Dense(self.include_head[-1], use_bias=True, name=f"bilma/dense_ex_final")(x), enc_output]
+            fin_output = [Dense(include_head[-1], use_bias=True, name=f"bilma/dense_ex_final")(x), enc_output]
     
     caption_model = Model(inputs=capt_inputs_ids, outputs=fin_output, name="bilma_model")
     return caption_model

tf_model.h5

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:f83fdad7da418dac337cc4df40cb630f3145ff66b48188148e899214539e2db5
+oid sha256:56d10c00c39895f8354e27f73dac3b6c4c0d3fcc8b146131c43be64f8868cfbe
 size 156875820