Update test.py

neuralmagic · Jul 12, 2023 · 4ce8d2a · 4ce8d2a
1 parent b830c6a
commit 4ce8d2a
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/examples/chat/test.py b/examples/chat/test.py
@@ -3,14 +3,14 @@
 
 
 ort_codegen = TextGenerationPipeline(
-    model_path="/home/mgoin/slow/nyann/deepsparse/codegen-350M-multi",
+    model_path="codegen-350M-multi",
     engine_type="onnxruntime",
     sequence_length=1024,
     num_tokens_to_generate=128,
     streamer=True)
 
 ds_codegen = TextGenerationPipeline(
-    model_path="/home/mgoin/slow/nyann/deepsparse/codegen-350M-multi",
+    model_path="codegen-350M-multi",
     engine_type="deepsparse",
     sequence_length=1024,
     num_tokens_to_generate=128,
@@ -79,4 +79,4 @@ def hello_world():
     print('Hello World!')
     print('World!')
     print('Hello World!')
-"""
+"""