Push model using huggingface_hub.

README.md

@@ -1,15 +1,10 @@
 ---
-language: en
-license: mit
 tags:
-- pytorch
-- pertTF
+- model_hub_mixin
+- pytorch_model_hub_mixin
 ---
 
-# pertTF tiny
-
-This is model files for pertTF large
-
-## Usage
-```python
-model = HFPerturbationTFModel.from_pretrained('weililab/pertTF-large')
+This model has been pushed to the Hub using the [PytorchModelHubMixin](https://huggingface.co/docs/huggingface_hub/package_reference/mixins#huggingface_hub.PyTorchModelHubMixin) integration:
+- Code: [More Information Needed]
+- Paper: [More Information Needed]
+- Docs: [More Information Needed]

config.json

@@ -8,7 +8,7 @@
   "dropout": 0.1,
   "ecs_threshold": 0.7,
   "explicit_zero_prob": true,
-  "fast_transformer_backend": "flash",
+  "fast_transformer_backend": "vanilla",
   "input_emb_style": "continuous",
   "mvc_decoder_style": "inner product",
   "n_bins": 51,

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:cb086ae64b651cbdf13564792d00a9626a7775b0f7e3194ec43301fefba0ad1c
-size 172008668
+oid sha256:4e89348c8f377a85c2d173bc5c29c5ff7e42e54108ea7a6e95782ee5f197e746
+size 172009132

training_config.json

@@ -80,7 +80,7 @@
   "ecs_threshold": 0.7,
   "explicit_zero_prob": true,
   "use_fast_transformer": true,
-  "fast_transformer_backend": "flash",
+  "fast_transformer_backend": "vanilla",
   "pre_norm": false,
   "pred_lochness_next": false,
   "ps_decoder2_nlayer": 3,