Professor commited on
Commit
702545c
·
verified ·
1 Parent(s): 7fb759b

Upload T5ForConditionalGeneration

Browse files
Files changed (2) hide show
  1. config.json +3 -1
  2. generation_config.json +1 -1
config.json CHANGED
@@ -23,6 +23,8 @@
23
  "output_past": true,
24
  "pad_token_id": 0,
25
  "quantization_config": {
 
 
26
  "bnb_4bit_compute_dtype": "float32",
27
  "bnb_4bit_quant_type": "fp4",
28
  "bnb_4bit_use_double_quant": false,
@@ -38,7 +40,7 @@
38
  "relative_attention_num_buckets": 32,
39
  "tie_word_embeddings": false,
40
  "torch_dtype": "float16",
41
- "transformers_version": "4.37.0",
42
  "use_cache": true,
43
  "vocab_size": 32100
44
  }
 
23
  "output_past": true,
24
  "pad_token_id": 0,
25
  "quantization_config": {
26
+ "_load_in_4bit": false,
27
+ "_load_in_8bit": true,
28
  "bnb_4bit_compute_dtype": "float32",
29
  "bnb_4bit_quant_type": "fp4",
30
  "bnb_4bit_use_double_quant": false,
 
40
  "relative_attention_num_buckets": 32,
41
  "tie_word_embeddings": false,
42
  "torch_dtype": "float16",
43
+ "transformers_version": "4.38.0.dev0",
44
  "use_cache": true,
45
  "vocab_size": 32100
46
  }
generation_config.json CHANGED
@@ -3,5 +3,5 @@
3
  "decoder_start_token_id": 0,
4
  "eos_token_id": 1,
5
  "pad_token_id": 0,
6
- "transformers_version": "4.37.0"
7
  }
 
3
  "decoder_start_token_id": 0,
4
  "eos_token_id": 1,
5
  "pad_token_id": 0,
6
+ "transformers_version": "4.38.0.dev0"
7
  }