Upload TFBilma

modeling_bilma.py

@@ -9,7 +9,7 @@ from typing import Dict
 import re
 import unicodedata
 
-from .configuration_bilma import BilmaConfig
+from configuration_bilma import BilmaConfig
 
 # copied from preprocessing.py
 BLANK = ' '
@@ -71,10 +71,13 @@ class TFBilma(TFPreTrainedModel):
         return sig
     
     
-    def call(self, capt_input):        
+    def call(self, inputs):        
         #if isinstance(tensor, dict) and len(tensor) == 0:
         #    return self.model(self.dummy_inputs)
-        return self.model(capt_input)
+        ins = tf.cast(inputs["input_ids"], tf.float32)
+        
+        output = {"logits":self.model(ins)}
+        return output
     
 
 # 

tf_model.h5

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:28582d643f857938d54653b1eb5481a6f4a8d68d0909b1af58371c55806e9048
+oid sha256:ae6a05836d990960fb2c4a9fa4616d18e0c396c2e075f85cc98d1ab8aecdc7a0
 size 156564220