Update README.md
Browse files
README.md
CHANGED
@@ -37,7 +37,7 @@ Then run the following code to infer.
|
|
37 |
import torch
|
38 |
from transformers import AutoModelForCausalLM, AutoTokenizer
|
39 |
|
40 |
-
model_name_or_path = "
|
41 |
tokenizer = AutoTokenizer.from_pretrained(model_name_or_path, use_fast=True, return_tensors="pt")
|
42 |
model = AutoModelForCausalLM.from_pretrained(
|
43 |
model_name_or_path, torch_dtype=torch.bfloat16,
|
|
|
37 |
import torch
|
38 |
from transformers import AutoModelForCausalLM, AutoTokenizer
|
39 |
|
40 |
+
model_name_or_path = "ClosedCharacter/Peach-9B-8k-Roleplay"
|
41 |
tokenizer = AutoTokenizer.from_pretrained(model_name_or_path, use_fast=True, return_tensors="pt")
|
42 |
model = AutoModelForCausalLM.from_pretrained(
|
43 |
model_name_or_path, torch_dtype=torch.bfloat16,
|