optimum-neuron-cache
/
neuronxcc-2.15.128.0+56dc5a86
/0_REGISTRY
/0.0.25.dev0
/training
/llama
/meta-llama
/Llama-3.2-1B-Instruct
/acf9d8ffb10c2cab4ccc.json
{"architectures": ["LlamaForCausalLM"], "attention_bias": false, "attention_dropout": 0.0, "bos_token_id": 128256, "eos_token_id": 128257, "head_dim": 64, "hidden_act": "silu", "hidden_size": 2048, "initializer_range": 0.02, "intermediate_size": 8192, "max_position_embeddings": 131072, "mlp_bias": false, "model_type": "llama", "neuron": {"compiler_version": "2.15.128.0+56dc5a86", "input_specs": {"chosen": ["<|im_start|>user\nExplain simply why 7 is the correct answer to: Nick has 35 quarters. 2/5 of the quarters are state quarters, and 50 percent of the state quarters are Pennsylvania. How many Pennsylvania state quarters does Nick have?. Explanation:<|im_end|>\n<|im_start|>assistant\nNick has 35 quarters, 2/5 of which are state quarters. This means there are 35 * 2/5 = 14 state quarters. Half of the state quarters are Pennsylvania, so there are $\\frac{1}{2} * 14 = 7$ Pennsylvania state quarters.\n\nConfidence: 100%.<|im_end|>\n"], "chosen_attention_mask": [1, 512], "chosen_input_ids": [1, 512], "chosen_labels": [1, 512], "prompt": ["Explain simply why 7 is the correct answer to: Nick has 35 quarters. 2/5 of the quarters are state quarters, and 50 percent of the state quarters are Pennsylvania. How many Pennsylvania state quarters does Nick have?. Explanation:"], "prompt_attention_mask": [1, 512], "prompt_input_ids": [1, 512], "question": ["Explain simply why 7 is the correct answer to: Nick has 35 quarters. 2/5 of the quarters are state quarters, and 50 percent of the state quarters are Pennsylvania. How many Pennsylvania state quarters does Nick have?. Explanation:"], "rejected": ["<|im_start|>user\nExplain simply why 7 is the correct answer to: Nick has 35 quarters. 2/5 of the quarters are state quarters, and 50 percent of the state quarters are Pennsylvania. How many Pennsylvania state quarters does Nick have?. Explanation:<|im_end|>\n<|im_start|>assistant\nNick has 35 quarters. Two-fifths of the quarters are state quarters, which means he has 17 state quarters (35 \u00f7 2 = 17.5). \nIf 50 percent of the state quarters are from Pennsylvania, then he has 8.5 Pennsylvania state quarters (17.5 \u00f7 2 = 8.75, and since 0.5 cannot be increased, 8.75 is rounded down to 8.5). Therefore, Nick has 85 state quarters (17 + 8.5 = 25.5, and since 50 percent of 25.5 is 12.75, the answer is rounded down to 7).<|im_end|>\n"], "rejected_attention_mask": [1, 512], "rejected_input_ids": [1, 512], "rejected_labels": [1, 512], "source": ["flan_v2_cot"]}, "model_class": "PeftModelForCausalLM", "num_neuron_cores_per_node": 2, "pipeline_parallel_size": 1, "precision": "bfloat16", "tensor_parallel_size": 1, "training": true}, "num_attention_heads": 32, "num_hidden_layers": 16, "num_key_value_heads": 8, "pad_token_id": 128257, "pretraining_tp": 1, "rms_norm_eps": 1e-05, "rope_scaling": {"factor": 32.0, "high_freq_factor": 4.0, "low_freq_factor": 1.0, "original_max_position_embeddings": 8192, "rope_type": "llama3"}, "rope_theta": 500000.0, "tie_word_embeddings": true, "torch_dtype": "bfloat16", "use_cache": true, "vocab_size": 128258} |