Bazsalanszky
commited on
Commit
•
72ad70f
1
Parent(s):
0df10f1
Update README.md
Browse files
README.md
CHANGED
@@ -31,8 +31,15 @@ Ez a modell NEM lett instrukciókra tanítva, valószínűleg nem fogja azokat k
|
|
31 |
# Load model directly
|
32 |
from transformers import AutoTokenizer, AutoModel
|
33 |
|
|
|
|
|
|
|
34 |
tokenizer = AutoTokenizer.from_pretrained("mistralai/Mistral-7B-v0.1")
|
35 |
-
model =
|
|
|
|
|
|
|
|
|
36 |
|
37 |
inputs = tokenizer("Magyarország\nFővárosa:", return_tensors = "pt").to("cpu")
|
38 |
|
|
|
31 |
# Load model directly
|
32 |
from transformers import AutoTokenizer, AutoModel
|
33 |
|
34 |
+
# Load model directly
|
35 |
+
from transformers import AutoTokenizer, AutoModelForCausalLM
|
36 |
+
|
37 |
tokenizer = AutoTokenizer.from_pretrained("mistralai/Mistral-7B-v0.1")
|
38 |
+
model = AutoModelForCausalLM.from_pretrained("mistralai/Mistral-7B-v0.1")
|
39 |
+
|
40 |
+
adapter_name = model.load_adapter("Bazsalanszky/Mistral-7b-0.1-hu")
|
41 |
+
|
42 |
+
model.active_adapters = adapter_name
|
43 |
|
44 |
inputs = tokenizer("Magyarország\nFővárosa:", return_tensors = "pt").to("cpu")
|
45 |
|