Fix architecture (#50)
Browse files- Fix architecture (905af2fbb8162cf423b311f87f5d57ec32677d30)
- config.json +2 -2
    	
        config.json
    CHANGED
    
    | @@ -2,7 +2,7 @@ | |
| 2 | 
             
              "apply_residual_connection_post_layernorm": false,
         | 
| 3 | 
             
              "attention_dropout": 0.0,
         | 
| 4 | 
             
              "architectures": [
         | 
| 5 | 
            -
                " | 
| 6 | 
             
              ],
         | 
| 7 | 
             
              "attention_softmax_in_fp32": true,
         | 
| 8 | 
             
              "bias_dropout_fusion": true,
         | 
| @@ -26,4 +26,4 @@ | |
| 26 | 
             
              "transformers_version": "4.20.0",
         | 
| 27 | 
             
              "use_cache": true,
         | 
| 28 | 
             
              "vocab_size": 250880
         | 
| 29 | 
            -
            }
         | 
|  | |
| 2 | 
             
              "apply_residual_connection_post_layernorm": false,
         | 
| 3 | 
             
              "attention_dropout": 0.0,
         | 
| 4 | 
             
              "architectures": [
         | 
| 5 | 
            +
                "BloomForCausalLM"
         | 
| 6 | 
             
              ],
         | 
| 7 | 
             
              "attention_softmax_in_fp32": true,
         | 
| 8 | 
             
              "bias_dropout_fusion": true,
         | 
|  | |
| 26 | 
             
              "transformers_version": "4.20.0",
         | 
| 27 | 
             
              "use_cache": true,
         | 
| 28 | 
             
              "vocab_size": 250880
         | 
| 29 | 
            +
            }
         | 

