Adding ONNX file of this model
#17
by
						
guychuk
	
							
						- opened
							
					
- onnx/config.json +47 -0
- onnx/generation_config.json +6 -0
- onnx/model.onnx +3 -0
    	
        onnx/config.json
    ADDED
    
    | @@ -0,0 +1,47 @@ | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | 
|  | |
| 1 | 
            +
            {
         | 
| 2 | 
            +
              "_attn_implementation_autoset": true,
         | 
| 3 | 
            +
              "_name_or_path": "onnx/distilbert/distilgpt2",
         | 
| 4 | 
            +
              "_num_labels": 1,
         | 
| 5 | 
            +
              "activation_function": "gelu_new",
         | 
| 6 | 
            +
              "architectures": [
         | 
| 7 | 
            +
                "GPT2LMHeadModel"
         | 
| 8 | 
            +
              ],
         | 
| 9 | 
            +
              "attn_pdrop": 0.1,
         | 
| 10 | 
            +
              "bos_token_id": 50256,
         | 
| 11 | 
            +
              "embd_pdrop": 0.1,
         | 
| 12 | 
            +
              "eos_token_id": 50256,
         | 
| 13 | 
            +
              "id2label": {
         | 
| 14 | 
            +
                "0": "LABEL_0"
         | 
| 15 | 
            +
              },
         | 
| 16 | 
            +
              "initializer_range": 0.02,
         | 
| 17 | 
            +
              "label2id": {
         | 
| 18 | 
            +
                "LABEL_0": 0
         | 
| 19 | 
            +
              },
         | 
| 20 | 
            +
              "layer_norm_epsilon": 1e-05,
         | 
| 21 | 
            +
              "model_type": "gpt2",
         | 
| 22 | 
            +
              "n_ctx": 1024,
         | 
| 23 | 
            +
              "n_embd": 768,
         | 
| 24 | 
            +
              "n_head": 12,
         | 
| 25 | 
            +
              "n_inner": null,
         | 
| 26 | 
            +
              "n_layer": 6,
         | 
| 27 | 
            +
              "n_positions": 1024,
         | 
| 28 | 
            +
              "reorder_and_upcast_attn": false,
         | 
| 29 | 
            +
              "resid_pdrop": 0.1,
         | 
| 30 | 
            +
              "scale_attn_by_inverse_layer_idx": false,
         | 
| 31 | 
            +
              "scale_attn_weights": true,
         | 
| 32 | 
            +
              "summary_activation": null,
         | 
| 33 | 
            +
              "summary_first_dropout": 0.1,
         | 
| 34 | 
            +
              "summary_proj_to_labels": true,
         | 
| 35 | 
            +
              "summary_type": "cls_index",
         | 
| 36 | 
            +
              "summary_use_proj": true,
         | 
| 37 | 
            +
              "task_specific_params": {
         | 
| 38 | 
            +
                "text-generation": {
         | 
| 39 | 
            +
                  "do_sample": true,
         | 
| 40 | 
            +
                  "max_length": 50
         | 
| 41 | 
            +
                }
         | 
| 42 | 
            +
              },
         | 
| 43 | 
            +
              "torch_dtype": "float32",
         | 
| 44 | 
            +
              "transformers_version": "4.49.0",
         | 
| 45 | 
            +
              "use_cache": true,
         | 
| 46 | 
            +
              "vocab_size": 50257
         | 
| 47 | 
            +
            }
         | 
    	
        onnx/generation_config.json
    ADDED
    
    | @@ -0,0 +1,6 @@ | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | 
|  | |
| 1 | 
            +
            {
         | 
| 2 | 
            +
              "_from_model_config": true,
         | 
| 3 | 
            +
              "bos_token_id": 50256,
         | 
| 4 | 
            +
              "eos_token_id": 50256,
         | 
| 5 | 
            +
              "transformers_version": "4.49.0"
         | 
| 6 | 
            +
            }
         | 
    	
        onnx/model.onnx
    ADDED
    
    | @@ -0,0 +1,3 @@ | |
|  | |
|  | |
|  | 
|  | |
| 1 | 
            +
            version https://git-lfs.github.com/spec/v1
         | 
| 2 | 
            +
            oid sha256:7d24bf72a519e96e5cc81478e46eabe456a808426175c5b816c85c9c150ab031
         | 
| 3 | 
            +
            size 327855549
         | 
