Update README.md
#2
by
ycat3
- opened
README.md
CHANGED
@@ -92,16 +92,14 @@ fn main() -> anyhow::Result<()> {
|
|
92 |
cache_subdir: "japanese-soseki-gpt2-1b/vocab".into(),
|
93 |
});
|
94 |
let vocab_resource_token = vocab_resource.clone();
|
95 |
-
let merges_resource = vocab_resource.clone();
|
96 |
let generate_config = GenerateConfig {
|
97 |
model_resource,
|
98 |
config_resource,
|
99 |
-
vocab_resource,
|
100 |
-
merges_resource, // not used
|
101 |
device: Device::Cpu,
|
102 |
repetition_penalty: 1.6,
|
103 |
min_length: 40,
|
104 |
-
max_length: 128,
|
105 |
do_sample: true,
|
106 |
early_stopping: true,
|
107 |
num_beams: 5,
|
|
|
92 |
cache_subdir: "japanese-soseki-gpt2-1b/vocab".into(),
|
93 |
});
|
94 |
let vocab_resource_token = vocab_resource.clone();
|
|
|
95 |
let generate_config = GenerateConfig {
|
96 |
model_resource,
|
97 |
config_resource,
|
98 |
+
vocab_resource,
|
|
|
99 |
device: Device::Cpu,
|
100 |
repetition_penalty: 1.6,
|
101 |
min_length: 40,
|
102 |
+
max_length: Some(128),
|
103 |
do_sample: true,
|
104 |
early_stopping: true,
|
105 |
num_beams: 5,
|