update code for inference
Browse filesslow tokenizer requirements
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 |
)
|