| { |
| "architectures": [ |
| "PrivateLLMForCausalLM" |
| ], |
| "auto_map": { |
| "AutoConfig": "configuration_private_llm.PrivateLLMConfig", |
| "AutoModel": "modeling_private_llm.PrivateLLMForCausalLM", |
| "AutoModelForCausalLM": "modeling_private_llm.PrivateLLMForCausalLM", |
| "AutoTokenizer": [ |
| "tokenization_private_llm.PrivateLLMTokenizer", |
| null |
| ] |
| }, |
| "bos_token_id": 258, |
| "custom_pipelines": { |
| "private-llm": { |
| "impl": "pipeline_private_llm.PrivateLLMPipeline", |
| "pt": [ |
| "AutoModelForCausalLM" |
| ] |
| } |
| }, |
| "eos_token_id": 257, |
| "execution_mode": "auto", |
| "generate_function": null, |
| "init_kwargs": {}, |
| "loader_function": null, |
| "model_class": null, |
| "model_type": "private_llm", |
| "pad_token_id": 256, |
| "private_output_includes_prompt": false, |
| "private_script": "private_LLM_model.py", |
| "subprocess_args": [], |
| "subprocess_prompt_mode": "stdin", |
| "subprocess_timeout": 300, |
| "tokenizer_class": "PrivateLLMTokenizer", |
| "tokenizer_vocab_file": "private_llm_tokenizer.json", |
| "torch_dtype": "float32", |
| "vocab_size": 260 |
| } |
|
|
