Update README.md

README.md

@@ -31,8 +31,15 @@ Ez a modell NEM lett instrukciókra tanítva, valószínűleg nem fogja azokat k
 # Load model directly
 from transformers import AutoTokenizer, AutoModel
 
+# Load model directly
+from transformers import AutoTokenizer, AutoModelForCausalLM
+
 tokenizer = AutoTokenizer.from_pretrained("mistralai/Mistral-7B-v0.1")
-model = AutoModel.from_pretrained("Bazsalanszky/Mistral-7b-0.1-hu")
+model = AutoModelForCausalLM.from_pretrained("mistralai/Mistral-7B-v0.1")
+
+adapter_name = model.load_adapter("Bazsalanszky/Mistral-7b-0.1-hu")
+
+model.active_adapters = adapter_name
 
 inputs = tokenizer("Magyarország\nFővárosa:", return_tensors = "pt").to("cpu")