ayanami-kitasan
/

code-pruner

Token Classification

context-pruning

Model card Files Files and versions

ayanami-kitasan commited on Jan 17

Commit

c5a550f

·

verified ·

1 Parent(s): 2ce9736

Update config.json

Files changed (1) hide show

config.json +1 -1

config.json CHANGED Viewed

@@ -9,7 +9,7 @@
   "dropout": 0.4,
   "early_layer_ratio": 0.25,
   "middle_layer_ratio": 0.5,
-  "model_type": "provence",
   "num_fusion_layers": 1,
   "num_heads": 8,
   "torch_dtype": "bfloat16",

   "dropout": 0.4,
   "early_layer_ratio": 0.25,
   "middle_layer_ratio": 0.5,
+  "model_type": "swepruner",
   "num_fusion_layers": 1,
   "num_heads": 8,
   "torch_dtype": "bfloat16",