Update README.md

README.md

@@ -9,7 +9,7 @@ model-index:
 - name: TinyLlama-v2ray
   results: []
 datasets:
-- TheBossLevel123/v2ray
+- theminji/v2ray
 library_name: transformers
 widget:
 - text: "<|im_start|>user\nWho are you?<|im_end|>\n<|im_start|>assistant"
@@ -21,7 +21,7 @@ widget:
 
 # TinyLlama-v2ray
 
-This model is a fine-tuned version of [TinyLlama/TinyLlama-1.1B-Chat-v0.6](https://huggingface.co/TinyLlama/TinyLlama-1.1B-Chat-v0.6) on the [TheBossLevel123/v2ray](https://huggingface.co/datasets/TheBossLevel123/v2ray) dataset.
+This model is a fine-tuned version of [TinyLlama/TinyLlama-1.1B-Chat-v0.6](https://huggingface.co/TinyLlama/TinyLlama-1.1B-Chat-v0.6) on the [theminji/v2ray](https://huggingface.co/datasets/theminji/v2ray) dataset.
 
 ## Model description
 Prompt format is as follows:
@@ -38,8 +38,8 @@ The model is intended to mimic the behavior of v2ray, so results will most likel
 import torch
 from transformers import pipeline, AutoTokenizer
 import re
-tokenizer = AutoTokenizer.from_pretrained("TheBossLevel123/TinyLlama-v2ray")
-pipe = pipeline("text-generation", model="TheBossLevel123/TinyLlama-v2ray", torch_dtype=torch.bfloat16, device_map="auto")
+tokenizer = AutoTokenizer.from_pretrained("theminji/TinyLlama-v2ray")
+pipe = pipeline("text-generation", model="theminji/TinyLlama-v2ray", torch_dtype=torch.bfloat16, device_map="auto")
 
 def formatted_prompt(prompt)-> str:
     return f"<|im_start|>user\n{prompt}<|im_end|>\n<|im_start|>assistant"