ClosedCharacter commited on
Commit
b5fc3ec
·
verified ·
1 Parent(s): 40b7b7f

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
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 = "losed-Character/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,
 
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,