Fix phi example (#1436)

* Fix phi example

* Remove the cuda mention.

---------

Co-authored-by: Laurent <laurent.mazare@gmail.com>
This commit is contained in:
niu tech
2023-12-15 14:01:10 +01:00
committed by GitHub
parent e60f9b5dfc
commit 79eab519fd

View File

@ -12,7 +12,7 @@ quantized variant.
For the v2 version.
```bash
$ cargo run --example phi --release cuda -- --prompt "def print_prime(n): " --model 2
$ cargo run --example phi --release -- --prompt "def print_prime(n): " --model 2
def print_prime(n):
if n <= 1:
print("Not a prime number")