bigscience
/

bigscience-small-testing

Text Generation

feature-extraction

text-generation-inference

Model card Files Files and versions

Younes Belkada commited on Jun 6, 2022

Commit

4d436da

·

1 Parent(s): 4fb31a5

Update config.json

Files changed (1) hide show

config.json +2 -2

config.json CHANGED Viewed

@@ -2,7 +2,7 @@
   "_name_or_path": "/home/younes/Desktop/Work/data/megatron-debug/",
   "apply_residual_connection_post_layernorm": false,
   "architectures": [
-    "BigScience176BModel"
   ],
   "attention_dropout": 0.1,
   "bias_dropout_fusion": true,
@@ -14,7 +14,7 @@
   "initializer_range": 0.02,
   "layer_norm_epsilon": 1e-05,
   "masked_softmax_fusion": true,
-  "model_type": "bigscience176b",
   "n_head": 8,
   "n_inner": null,
   "n_layer": 2,

   "_name_or_path": "/home/younes/Desktop/Work/data/megatron-debug/",
   "apply_residual_connection_post_layernorm": false,
   "architectures": [
+    "BloomModel"
   ],
   "attention_dropout": 0.1,
   "bias_dropout_fusion": true,
   "initializer_range": 0.02,
   "layer_norm_epsilon": 1e-05,
   "masked_softmax_fusion": true,
+  "model_type": "bloom",
   "n_head": 8,
   "n_inner": null,
   "n_layer": 2,