Update README.md
Browse files
README.md
CHANGED
|
@@ -201,12 +201,12 @@ import torch
|
|
| 201 |
|
| 202 |
# Load model and tokenizer
|
| 203 |
model = AutoModelForCausalLM.from_pretrained(
|
| 204 |
-
"
|
| 205 |
trust_remote_code=True,
|
| 206 |
torch_dtype=torch.bfloat16,
|
| 207 |
device_map="auto"
|
| 208 |
)
|
| 209 |
-
tokenizer = AutoTokenizer.from_pretrained("
|
| 210 |
|
| 211 |
# Generate text
|
| 212 |
messages = [{"role": "user", "content": "Explain the waterfall model in software engineering."}]
|
|
|
|
| 201 |
|
| 202 |
# Load model and tokenizer
|
| 203 |
model = AutoModelForCausalLM.from_pretrained(
|
| 204 |
+
"NoesisLab/Asterisk-Pi",
|
| 205 |
trust_remote_code=True,
|
| 206 |
torch_dtype=torch.bfloat16,
|
| 207 |
device_map="auto"
|
| 208 |
)
|
| 209 |
+
tokenizer = AutoTokenizer.from_pretrained("NoesisLab/Asterisk-Pi")
|
| 210 |
|
| 211 |
# Generate text
|
| 212 |
messages = [{"role": "user", "content": "Explain the waterfall model in software engineering."}]
|