Update app.py
Browse files
app.py
CHANGED
@@ -19,11 +19,12 @@ MAX_MAX_NEW_TOKENS = 1024
|
|
19 |
DEFAULT_MAX_NEW_TOKENS = 200
|
20 |
|
21 |
# Setup the engine
|
22 |
-
|
|
|
23 |
task="text-generation",
|
24 |
model_path=MODEL_ID,
|
25 |
sequence_length=MAX_MAX_NEW_TOKENS,
|
26 |
-
prompt_sequence_length=
|
27 |
num_cores=8,
|
28 |
)
|
29 |
|
|
|
19 |
DEFAULT_MAX_NEW_TOKENS = 200
|
20 |
|
21 |
# Setup the engine
|
22 |
+
from deepsparse.legacy import Pipeline
|
23 |
+
pipe = Pipeline.create(
|
24 |
task="text-generation",
|
25 |
model_path=MODEL_ID,
|
26 |
sequence_length=MAX_MAX_NEW_TOKENS,
|
27 |
+
prompt_sequence_length=8,
|
28 |
num_cores=8,
|
29 |
)
|
30 |
|