Update README.md
Browse files
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=
|
|
|
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 = """
|