ydshieh
/

tf_custom_model_example

Model card Files Files and versions

ydshieh commited on Dec 8, 2023

Commit

7350e48

·

1 Parent(s): c6bc350

initial commit

Files changed (1) hide show

config.json +2 -1

config.json CHANGED Viewed

@@ -3,6 +3,7 @@
   "model_type": "my_model",
   "auto_map": {
     "AutoConfig": "configuration_my_model.MyModelConfig",
-    "AutoModel": "modeling_my_model.MyModel"
   }
 }

   "model_type": "my_model",
   "auto_map": {
     "AutoConfig": "configuration_my_model.MyModelConfig",
+    "AutoModel": "modeling_my_model.MyModel",
+    "TFAutoModel": "modeling_tf_my_model.TFMyModel"
   }
 }