kdf/python-docstring-generation

1{2  "_name_or_path": "drive/MyDrive/AI/docstring/py_docstring_model_220000_e1",3  "activation_function": "gelu_new",4  "architectures": [5    "CodeGenForCausalLM"6  ],7  "attn_pdrop": 0.0,8  "bos_token_id": 1,9  "embd_pdrop": 0.0,10  "eos_token_id": 2,11  "gradient_checkpointing": false,12  "initializer_range": 0.02,13  "layer_norm_epsilon": 1e-05,14  "model_type": "codegen",15  "n_ctx": 2048,16  "n_embd": 1024,17  "n_head": 16,18  "n_inner": null,19  "n_layer": 20,20  "n_positions": 2048,21  "resid_pdrop": 0.0,22  "rotary_dim": 32,23  "scale_attn_weights": true,24  "summary_activation": null,25  "summary_first_dropout": 0.1,26  "summary_proj_to_labels": true,27  "summary_type": "cls_index",28  "summary_use_proj": true,29  "task_specific_params": {30    "text-generation": {31      "do_sample": true,32      "max_length": 50,33      "temperature": 1.034    }35  },36  "tie_word_embeddings": false,37  "tokenizer_class": "GPT2Tokenizer",38  "torch_dtype": "float32",39  "transformers_version": "4.22.0.dev0",40  "use_cache": true,41  "vocab_size": 5120042}43