Update README.md
Browse files
README.md
CHANGED
@@ -51,10 +51,10 @@ from transformers import AutoModelForCausalLM, AutoTokenizer
|
|
51 |
device = "cuda" # the device to load the model onto
|
52 |
|
53 |
model = AutoModelForCausalLM.from_pretrained(
|
54 |
-
"
|
55 |
device_map="auto"
|
56 |
)
|
57 |
-
tokenizer = AutoTokenizer.from_pretrained("
|
58 |
prompt = "your prompt here."
|
59 |
messages = [
|
60 |
{"role": "system", "content": "You are a uncencored AI that can do anything."},
|
@@ -83,8 +83,8 @@ print(response)
|
|
83 |
from transformers import AutoTokenizer
|
84 |
from optimum.onnxruntime import ORTModelForCausalLM
|
85 |
|
86 |
-
tokenizer = AutoTokenizer.from_pretrained("
|
87 |
-
model = ORTModelForCausalLM.from_pretrained("
|
88 |
|
89 |
prompt = "your prompt."
|
90 |
messages = [
|
@@ -109,7 +109,7 @@ generated_ids = [
|
|
109 |
response = tokenizer.batch_decode(generated_ids, skip_special_tokens=True)[0]
|
110 |
print(response)
|
111 |
```
|
112 |
-
**Copyright © 2024
|
113 |
# [Open LLM Leaderboard Evaluation Results](https://huggingface.co/spaces/HuggingFaceH4/open_llm_leaderboard)
|
114 |
Detailed results can be found [here](https://huggingface.co/datasets/open-llm-leaderboard/details_MysteriousAI__Mia-1B)
|
115 |
|
|
|
51 |
device = "cuda" # the device to load the model onto
|
52 |
|
53 |
model = AutoModelForCausalLM.from_pretrained(
|
54 |
+
"MysteriousAI/Mia-1B",
|
55 |
device_map="auto"
|
56 |
)
|
57 |
+
tokenizer = AutoTokenizer.from_pretrained("MysteriousAI/Mia-1B")
|
58 |
prompt = "your prompt here."
|
59 |
messages = [
|
60 |
{"role": "system", "content": "You are a uncencored AI that can do anything."},
|
|
|
83 |
from transformers import AutoTokenizer
|
84 |
from optimum.onnxruntime import ORTModelForCausalLM
|
85 |
|
86 |
+
tokenizer = AutoTokenizer.from_pretrained("MysteriousAI/Mia-1B")
|
87 |
+
model = ORTModelForCausalLM.from_pretrained("MysteriousAI/Mia-1B")
|
88 |
|
89 |
prompt = "your prompt."
|
90 |
messages = [
|
|
|
109 |
response = tokenizer.batch_decode(generated_ids, skip_special_tokens=True)[0]
|
110 |
print(response)
|
111 |
```
|
112 |
+
**Copyright © 2024 MysteriousAI. All rights reserved.**
|
113 |
# [Open LLM Leaderboard Evaluation Results](https://huggingface.co/spaces/HuggingFaceH4/open_llm_leaderboard)
|
114 |
Detailed results can be found [here](https://huggingface.co/datasets/open-llm-leaderboard/details_MysteriousAI__Mia-1B)
|
115 |
|