Push model using huggingface_hub.

README.md

@@ -1,7 +1,7 @@
 ---
 tags:
-- model_hub_mixin
 - pytorch_model_hub_mixin
+- model_hub_mixin
 ---
 
 This model has been pushed to the Hub using the [PytorchModelHubMixin](https://huggingface.co/docs/huggingface_hub/package_reference/mixins#huggingface_hub.PyTorchModelHubMixin) integration:

config.json

@@ -1,5 +1,5 @@
 {
-  "device": "cpu",
+  "device": "cuda",
   "dropout": 0.2,
   "n_blocks": 3,
   "transformer_config": {
@@ -13,6 +13,7 @@
     "batch_size": 4,
     "block_size": 134,
     "config_block": "msm",
+    "device": "cuda",
     "n_embd": 164
   }
 }

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:dc791baa75400b0b9baaf621cd58940dea980630ac66d84caba8ed1b7d223805
+oid sha256:9d51116eac3d7a060a4cc5aa4e8f3e74eeac861627ce2e467eb3d12d84ebfd47
 size 120894896