| # config.json | |
| { | |
| "_name_or_path": "EleutherAI/gpt-j-6B", | |
| "architectures": ["GPTJForCausalLM"], | |
| "model_type": "gptj", | |
| "torch_dtype": "float16", | |
| "quantization_config": { | |
| "load_in_4bit": true, | |
| "bnb_4bit_quant_type": "nf4", | |
| "bnb_4bit_compute_dtype": "float16", | |
| "bnb_4bit_use_double_quant": true | |
| } | |
| } |