optimum-internal-testing
/

tiny_random_bert_neuronx

Commit

91b8355

verified ·

1 Parent(s): 77653e6

Upload config.json with huggingface_hub

Files changed (1) hide show

config.json CHANGED Viewed

@@ -1,6 +1,6 @@
 {
   "_attn_implementation_autoset": true,
-  "_name_or_path": "/tmp/tmpyprqhas7",
   "architectures": [
     "BertModel"
   ],
@@ -18,7 +18,7 @@
     "auto_cast": null,
     "auto_cast_type": null,
     "compiler_type": "neuronx-cc",
-    "compiler_version": "2.15.143.0+e39249ad",
     "disable_fallback": false,
     "disable_fast_relayout": false,
     "dynamic_batch_size": false,

 {
   "_attn_implementation_autoset": true,
+  "_name_or_path": "/tmp/tmp8yseg49e",
   "architectures": [
     "BertModel"
   ],
     "auto_cast": null,
     "auto_cast_type": null,
     "compiler_type": "neuronx-cc",
+    "compiler_version": "2.16.372.0+4a9b2326",
     "disable_fallback": false,
     "disable_fast_relayout": false,
     "dynamic_batch_size": false,