{ "_entry_class": "SingleModelCacheEntry", "_model_id": "llamafactory/tiny-random-Llama-3", "_task": "text-generation", "architectures": [ "LlamaForCausalLM" ], "attention_bias": false, "attention_dropout": 0.0, "head_dim": 4, "hidden_act": "silu", "hidden_size": 16, "initializer_range": 0.02, "intermediate_size": 64, "max_position_embeddings": 131072, "mlp_bias": false, "model_type": "llama", "neuron": { "_serialized_key": "NxDNeuronConfig", "async_mode": false, "attn_kernel_enabled": false, "batch_size": 1, "capacity_factor": null, "cc_pipeline_tiling_factor": 2, "checkpoint_id": "llamafactory/tiny-random-Llama-3", "checkpoint_revision": "bf2a2e3bf199ad2ee96f02a3c00246c608db22a8", "continuous_batching": false, "enable_bucketing": false, "ep_degree": 1, "flash_decoding_enabled": false, "fused_qkv": true, "glu_mlp": true, "is_chunked_prefill": false, "local_ranks_size": 2, "logical_nc_config": 1, "max_batch_size": 1, "max_context_length": 100, "max_topk": 256, "mlp_kernel_enabled": false, "mlp_kernel_fuse_residual_add": false, "n_active_tokens": 100, "neuronxcc_version": "2.19.8089.0+8ab9f450", "num_cores_per_group": 1, "on_device_sampling": true, "optimum_neuron_version": "0.3.0.dev2", "output_logits": false, "padding_side": "right", "pp_degree": 1, "qk_layernorm": false, "qkv_kernel_enabled": false, "rpl_reduce_dtype": "float16", "sequence_length": 100, "sequence_parallel_enabled": false, "speculation_length": 0, "start_rank_id": 0, "target": null, "torch_dtype": "float16", "tp_degree": 2, "vocab_parallel": false }, "num_attention_heads": 4, "num_hidden_layers": 2, "num_key_value_heads": 4, "pretraining_tp": 1, "rms_norm_eps": 1e-05, "rope_scaling": { "factor": 8.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": false, "use_cache": true, "vocab_size": 128256 }