Update README.md

#2
by ycat3 - opened
Files changed (1) hide show
  1. README.md +2 -4
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,