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 |
|