pszemraj commited on
Commit
9ae3f9c
1 Parent(s): f583010

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -1
README.md CHANGED
@@ -44,7 +44,8 @@ model_tag = "pszemraj/opt-350m-email-generation"
44
  generator = pipeline(
45
  'text-generation',
46
  model=model_tag,
47
- do_sample=True,
 
48
  )
49
 
50
  prompt = """
 
44
  generator = pipeline(
45
  'text-generation',
46
  model=model_tag,
47
+ do_sample=False,
48
+ early_stopping=True,
49
  )
50
 
51
  prompt = """