sanchit-gandhi
		
	commited on
		
		
					Commit 
							
							·
						
						2d276f0
	
1
								Parent(s):
							
							6fcee57
								
Add Flax weights
Browse files- config.json +1 -1
- flax_model.msgpack +3 -0
- generation_config.json +14 -12
    	
        config.json
    CHANGED
    
    | @@ -136,7 +136,7 @@ | |
| 136 | 
             
                50362
         | 
| 137 | 
             
              ],
         | 
| 138 | 
             
              "torch_dtype": "float32",
         | 
| 139 | 
            -
              "transformers_version": "4. | 
| 140 | 
             
              "use_cache": true,
         | 
| 141 | 
             
              "vocab_size": 51865
         | 
| 142 | 
             
            }
         | 
|  | |
| 136 | 
             
                50362
         | 
| 137 | 
             
              ],
         | 
| 138 | 
             
              "torch_dtype": "float32",
         | 
| 139 | 
            +
              "transformers_version": "4.27.0.dev0",
         | 
| 140 | 
             
              "use_cache": true,
         | 
| 141 | 
             
              "vocab_size": 51865
         | 
| 142 | 
             
            }
         | 
    	
        flax_model.msgpack
    ADDED
    
    | @@ -0,0 +1,3 @@ | |
|  | |
|  | |
|  | 
|  | |
| 1 | 
            +
            version https://git-lfs.github.com/spec/v1
         | 
| 2 | 
            +
            oid sha256:c375de483cc3fef3050ee369779c112f177e99ef010bb52cc9a2e72f9d7b0a9d
         | 
| 3 | 
            +
            size 966956827
         | 
    	
        generation_config.json
    CHANGED
    
    | @@ -1,5 +1,4 @@ | |
| 1 | 
             
            {
         | 
| 2 | 
            -
              "_from_model_config": false,
         | 
| 3 | 
             
              "begin_suppress_tokens": [
         | 
| 4 | 
             
                220,
         | 
| 5 | 
             
                50257
         | 
| @@ -8,13 +7,15 @@ | |
| 8 | 
             
              "decoder_start_token_id": 50258,
         | 
| 9 | 
             
              "eos_token_id": 50257,
         | 
| 10 | 
             
              "forced_decoder_ids": [
         | 
| 11 | 
            -
                [ | 
| 12 | 
            -
                   | 
| 13 | 
            -
             | 
|  | |
|  | |
|  | |
| 14 | 
             
                  50359
         | 
| 15 | 
             
                ]
         | 
| 16 | 
             
              ],
         | 
| 17 | 
            -
              "return_timestamps":false,
         | 
| 18 | 
             
              "is_multilingual": true,
         | 
| 19 | 
             
              "lang_to_id": {
         | 
| 20 | 
             
                "<|af|>": 50327,
         | 
| @@ -47,8 +48,8 @@ | |
| 47 | 
             
                "<|gu|>": 50333,
         | 
| 48 | 
             
                "<|haw|>": 50352,
         | 
| 49 | 
             
                "<|ha|>": 50354,
         | 
|  | |
| 50 | 
             
                "<|hi|>": 50276,
         | 
| 51 | 
            -
                "<|he|>": 50279,    
         | 
| 52 | 
             
                "<|hr|>": 50291,
         | 
| 53 | 
             
                "<|ht|>": 50339,
         | 
| 54 | 
             
                "<|hu|>": 50286,
         | 
| @@ -118,13 +119,10 @@ | |
| 118 | 
             
                "<|zh|>": 50260
         | 
| 119 | 
             
              },
         | 
| 120 | 
             
              "max_initial_timestamp_index": 1,
         | 
| 121 | 
            -
              "no_timestamps_token_id": 50363,
         | 
| 122 | 
            -
              "task_to_id": {
         | 
| 123 | 
            -
                "transcribe": 50359,
         | 
| 124 | 
            -
                "translate": 50358
         | 
| 125 | 
            -
              },
         | 
| 126 | 
             
              "max_length": 448,
         | 
|  | |
| 127 | 
             
              "pad_token_id": 50256,
         | 
|  | |
| 128 | 
             
              "suppress_tokens": [
         | 
| 129 | 
             
                1,
         | 
| 130 | 
             
                2,
         | 
| @@ -213,5 +211,9 @@ | |
| 213 | 
             
                50361,
         | 
| 214 | 
             
                50362
         | 
| 215 | 
             
              ],
         | 
| 216 | 
            -
              " | 
|  | |
|  | |
|  | |
|  | |
| 217 | 
             
            }
         | 
|  | |
| 1 | 
             
            {
         | 
|  | |
| 2 | 
             
              "begin_suppress_tokens": [
         | 
| 3 | 
             
                220,
         | 
| 4 | 
             
                50257
         | 
|  | |
| 7 | 
             
              "decoder_start_token_id": 50258,
         | 
| 8 | 
             
              "eos_token_id": 50257,
         | 
| 9 | 
             
              "forced_decoder_ids": [
         | 
| 10 | 
            +
                [
         | 
| 11 | 
            +
                  1,
         | 
| 12 | 
            +
                  null
         | 
| 13 | 
            +
                ],
         | 
| 14 | 
            +
                [
         | 
| 15 | 
            +
                  2,
         | 
| 16 | 
             
                  50359
         | 
| 17 | 
             
                ]
         | 
| 18 | 
             
              ],
         | 
|  | |
| 19 | 
             
              "is_multilingual": true,
         | 
| 20 | 
             
              "lang_to_id": {
         | 
| 21 | 
             
                "<|af|>": 50327,
         | 
|  | |
| 48 | 
             
                "<|gu|>": 50333,
         | 
| 49 | 
             
                "<|haw|>": 50352,
         | 
| 50 | 
             
                "<|ha|>": 50354,
         | 
| 51 | 
            +
                "<|he|>": 50279,
         | 
| 52 | 
             
                "<|hi|>": 50276,
         | 
|  | |
| 53 | 
             
                "<|hr|>": 50291,
         | 
| 54 | 
             
                "<|ht|>": 50339,
         | 
| 55 | 
             
                "<|hu|>": 50286,
         | 
|  | |
| 119 | 
             
                "<|zh|>": 50260
         | 
| 120 | 
             
              },
         | 
| 121 | 
             
              "max_initial_timestamp_index": 1,
         | 
|  | |
|  | |
|  | |
|  | |
|  | |
| 122 | 
             
              "max_length": 448,
         | 
| 123 | 
            +
              "no_timestamps_token_id": 50363,
         | 
| 124 | 
             
              "pad_token_id": 50256,
         | 
| 125 | 
            +
              "return_timestamps": false,
         | 
| 126 | 
             
              "suppress_tokens": [
         | 
| 127 | 
             
                1,
         | 
| 128 | 
             
                2,
         | 
|  | |
| 211 | 
             
                50361,
         | 
| 212 | 
             
                50362
         | 
| 213 | 
             
              ],
         | 
| 214 | 
            +
              "task_to_id": {
         | 
| 215 | 
            +
                "transcribe": 50359,
         | 
| 216 | 
            +
                "translate": 50358
         | 
| 217 | 
            +
              },
         | 
| 218 | 
            +
              "transformers_version": "4.27.0.dev0"
         | 
| 219 | 
             
            }
         | 
