neuronxcc-2.15.128.0+56dc5a86/0_REGISTRY/0.0.25.dev0/inference/levit/hf-internal-testing/tiny-random-LevitModel/9e9a96f0a2b8d033370e.json

{"architectures": ["LevitModel"], "attention_ratio": [2, 2, 2], "depths": [2, 3, 4], "down_ops": [["Subsample", 16, 8, 4, 2, 2], ["Subsample", 16, 16, 4, 2, 2], [""]], "drop_path_rate": 0, "hidden_sizes": [128, 256, 384], "image_size": 64, "initializer_range": 0.02, "kernel_size": 3, "key_dim": [16, 16, 16], "mlp_ratio": [2, 2, 2], "model_type": "levit", "neuron": {"auto_cast": null, "auto_cast_type": null, "compiler_type": "neuronx-cc", "compiler_version": "2.15.128.0+56dc5a86", "disable_fallback": false, "disable_fast_relayout": false, "dynamic_batch_size": true, "inline_weights_to_neff": true, "optlevel": "2", "output_attentions": false, "output_hidden_states": false, "static_batch_size": 1, "static_image_size": 64, "static_num_channels": 3, "static_patch_size": 16}, "num_attention_heads": [4, 6, 8], "num_channels": 3, "padding": 1, "patch_size": 16, "stride": 2, "task": "image-classification"}