Text Generation
Transformers
Safetensors
PyTorch
nemotron_h
nvidia
unsloth
conversational
custom_code
8-bit precision
danielhanchen commited on
Commit
f27f0f4
·
verified ·
1 Parent(s): 694b35c

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +3 -8
config.json CHANGED
@@ -12,7 +12,6 @@
12
  "bos_token_id": 1,
13
  "chunk_size": 128,
14
  "conv_kernel": 4,
15
- "torch_dtype": "bfloat16",
16
  "eos_token_id": 2,
17
  "expand": 2,
18
  "head_dim": 128,
@@ -44,7 +43,7 @@
44
  "num_hidden_layers": 52,
45
  "num_key_value_heads": 2,
46
  "num_logits_to_keep": 1,
47
- "pad_token_id": 999,
48
  "partial_rotary_factor": 1.0,
49
  "rescale_prenorm_residual": true,
50
  "residual_in_fp32": false,
@@ -54,15 +53,11 @@
54
  "ssm_state_size": 128,
55
  "tie_word_embeddings": false,
56
  "time_step_floor": 0.0001,
57
- "time_step_limit": [
58
- 0.0,
59
- Infinity
60
- ],
61
  "time_step_max": 0.1,
62
  "time_step_min": 0.001,
63
  "topk_group": 1,
64
- "transformers_version": "4.57.6",
65
- "unsloth_fixed": true,
66
  "use_bias": false,
67
  "use_cache": true,
68
  "use_conv_bias": true,
 
12
  "bos_token_id": 1,
13
  "chunk_size": 128,
14
  "conv_kernel": 4,
 
15
  "eos_token_id": 2,
16
  "expand": 2,
17
  "head_dim": 128,
 
43
  "num_hidden_layers": 52,
44
  "num_key_value_heads": 2,
45
  "num_logits_to_keep": 1,
46
+ "pad_token_id": 0,
47
  "partial_rotary_factor": 1.0,
48
  "rescale_prenorm_residual": true,
49
  "residual_in_fp32": false,
 
53
  "ssm_state_size": 128,
54
  "tie_word_embeddings": false,
55
  "time_step_floor": 0.0001,
 
 
 
 
56
  "time_step_max": 0.1,
57
  "time_step_min": 0.001,
58
  "topk_group": 1,
59
+ "torch_dtype": "bfloat16",
60
+ "transformers_version": "4.53.2",
61
  "use_bias": false,
62
  "use_cache": true,
63
  "use_conv_bias": true,