ybelkada commited on
Commit
072e5dc
1 Parent(s): ea465be

Upload BloomForCausalLM

Browse files
Files changed (2) hide show
  1. config.json +1 -1
  2. quantization_config.json +8 -0
config.json CHANGED
@@ -1,5 +1,5 @@
1
  {
2
- "_name_or_path": "./test_bloom_8bit",
3
  "apply_residual_connection_post_layernorm": false,
4
  "architectures": [
5
  "BloomForCausalLM"
 
1
  {
2
+ "_name_or_path": "bigscience/bloom-560m",
3
  "apply_residual_connection_post_layernorm": false,
4
  "architectures": [
5
  "BloomForCausalLM"
quantization_config.json ADDED
@@ -0,0 +1,8 @@
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_from_model_config": false,
3
+ "llm_int8_enable_fp32_cpu_offload": false,
4
+ "llm_int8_skip_modules": null,
5
+ "llm_int8_threshold": 6.0,
6
+ "load_in_8bit": true,
7
+ "transformers_version": "4.28.0.dev0"
8
+ }