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")