Update README.md
Browse files
README.md
CHANGED
@@ -51,7 +51,7 @@ import torch
|
|
51 |
from transformers import pipeline
|
52 |
|
53 |
generate_text = pipeline(
|
54 |
-
model="
|
55 |
torch_dtype="auto",
|
56 |
trust_remote_code=True,
|
57 |
use_fast=False,
|
|
|
51 |
from transformers import pipeline
|
52 |
|
53 |
generate_text = pipeline(
|
54 |
+
model="PAIXAI/Astrid-LLama-7B",
|
55 |
torch_dtype="auto",
|
56 |
trust_remote_code=True,
|
57 |
use_fast=False,
|