Ray2333 commited on
Commit
d2d8717
1 Parent(s): 865fb0d

Upload config.json

Browse files
Files changed (1) hide show
  1. config.json +3 -3
config.json CHANGED
@@ -1,5 +1,5 @@
1
  {
2
- "_name_or_path": "/Users/yangrui/Desktop/SMART/gpt2-large-harmless-reward_model",
3
  "activation_function": "gelu_new",
4
  "architectures": [
5
  "GPT2ForSequenceClassification"
@@ -38,8 +38,8 @@
38
  "max_length": 50
39
  }
40
  },
41
- "torch_dtype": "float32",
42
- "transformers_version": "4.35.2",
43
  "use_cache": true,
44
  "vocab_size": 50257
45
  }
 
1
  {
2
+ "_name_or_path": "Ray2333/gpt2-large-harmless-reward_model",
3
  "activation_function": "gelu_new",
4
  "architectures": [
5
  "GPT2ForSequenceClassification"
 
38
  "max_length": 50
39
  }
40
  },
41
+ "torch_dtype": "bfloat16",
42
+ "transformers_version": "4.36.2",
43
  "use_cache": true,
44
  "vocab_size": 50257
45
  }