safetensors, config changes
#12
by
itay-levy
- opened
- config.json +1 -2
- configuration_decicoder.py +0 -1
- pytorch_model.bin → model.safetensors +2 -2
config.json
CHANGED
|
@@ -1,6 +1,6 @@
|
|
| 1 |
{
|
| 2 |
"architectures": [
|
| 3 |
-
"
|
| 4 |
],
|
| 5 |
"auto_map": {
|
| 6 |
"AutoConfig": "configuration_decicoder.DeciCoderConfig",
|
|
@@ -13,7 +13,6 @@
|
|
| 13 |
"initializer_range": 0.02,
|
| 14 |
"intermediate_size": 5888,
|
| 15 |
"max_position_embeddings": 2048,
|
| 16 |
-
"model_type": "llama",
|
| 17 |
"num_attention_heads": 32,
|
| 18 |
"num_hidden_layers": 20,
|
| 19 |
"num_key_value_heads": 4,
|
|
|
|
| 1 |
{
|
| 2 |
"architectures": [
|
| 3 |
+
"DeciCoderForCausalLM"
|
| 4 |
],
|
| 5 |
"auto_map": {
|
| 6 |
"AutoConfig": "configuration_decicoder.DeciCoderConfig",
|
|
|
|
| 13 |
"initializer_range": 0.02,
|
| 14 |
"intermediate_size": 5888,
|
| 15 |
"max_position_embeddings": 2048,
|
|
|
|
| 16 |
"num_attention_heads": 32,
|
| 17 |
"num_hidden_layers": 20,
|
| 18 |
"num_key_value_heads": 4,
|
configuration_decicoder.py
CHANGED
|
@@ -33,7 +33,6 @@ class DeciCoderConfig(LlamaConfig):
|
|
| 33 |
|
| 34 |
|
| 35 |
```"""
|
| 36 |
-
model_type = "llama"
|
| 37 |
keys_to_ignore_at_inference = ["past_key_values"]
|
| 38 |
|
| 39 |
def __init__(
|
|
|
|
| 33 |
|
| 34 |
|
| 35 |
```"""
|
|
|
|
| 36 |
keys_to_ignore_at_inference = ["past_key_values"]
|
| 37 |
|
| 38 |
def __init__(
|
pytorch_model.bin → model.safetensors
RENAMED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
-
size
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:510256faa3d388cad1dcbc30c39d32f9289410a399f4c0435bec27ec135c6f0f
|
| 3 |
+
size 2227364400
|