From 1c89d6d337d7b55f05784c031a11959dc76d36c0 Mon Sep 17 00:00:00 2001 From: Ryan Hill Date: Mon, 26 Feb 2024 22:33:16 -0800 Subject: [PATCH] Restore original model path --- examples/phi2/phi2-streaming.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/examples/phi2/phi2-streaming.py b/examples/phi2/phi2-streaming.py index e169bbf81..688fa324a 100644 --- a/examples/phi2/phi2-streaming.py +++ b/examples/phi2/phi2-streaming.py @@ -3,7 +3,7 @@ print("Loading model...") # The first argument is the name of the folder containing the model files -model=og.Model("model", og.DeviceType.CPU) +model=og.Model("example-models/phi2-int4-cpu", og.DeviceType.CPU) print("Model loaded") tokenizer=model.create_tokenizer() print("Tokenizer created")