pszemraj commited on
Commit
280daa0
1 Parent(s): ab14fc6

update code for inference

Browse files

slow tokenizer requirements

Files changed (1) hide show
  1. README.md +1 -0
README.md CHANGED
@@ -47,6 +47,7 @@ model_tag = "pszemraj/opt-350m-email-generation"
47
  generator = pipeline(
48
  'text-generation',
49
  model=model_tag,
 
50
  do_sample=False,
51
  early_stopping=True,
52
  )
 
47
  generator = pipeline(
48
  'text-generation',
49
  model=model_tag,
50
+ use_fast=False,
51
  do_sample=False,
52
  early_stopping=True,
53
  )