Update README.md

README.md

@@ -4,7 +4,7 @@ datasets:
 - tatsu-lab/alpaca
 ---
 
-This repo contains a low-rank adapter for LLaMA-13b fit on the Stanford Alpaca dataset.
+This repo contains a low-rank adapter for LLaMA-7b fit on the Stanford Alpaca dataset translated to Polish language.
 
 ### How to use (8-bit)
 
@@ -13,17 +13,14 @@ import torch
 from peft import PeftModel
 from transformers import LlamaTokenizer, LlamaForCausalLM, GenerationConfig
 
-tokenizer = LlamaTokenizer.from_pretrained("decapoda-research/llama-13b-hf")
-model = LlamaForCausalLM.from_pretrained(
-        "decapoda-research/llama-13b-hf",
-        load_in_8bit=True,
-        torch_dtype=torch.float16,
-        device_map="auto",
-)
-model = PeftModel.from_pretrained(
-    model, "baruga/alpaca-lora-13b",
-    torch_dtype=torch.float16
+tokenizer = LLaMATokenizer.from_pretrained("decapoda-research/llama-13b-hf")
+
+model = LLaMAForCausalLM.from_pretrained(
+    "decapoda-research/llama-13b-hf",
+    load_in_8bit=True,
+    device_map="auto",
 )
-```
 
-For further information, check out this Github repo: https://github.com/tloen/alpaca-lora.
+
+model = PeftModel.from_pretrained(model, "lora-alpaca-pl4-13")
+```