Update README.md
Browse files
README.md
CHANGED
@@ -70,9 +70,9 @@ Use the code below to get started with the model.
|
|
70 |
from peft import PeftModel, PeftConfig
|
71 |
from transformers import AutoModelForCausalLM
|
72 |
|
73 |
-
config = PeftConfig.from_pretrained("aoxo/
|
74 |
base_model = AutoModelForCausalLM.from_pretrained("unsloth/Meta-Llama-3.1-8B-bnb-4bit")
|
75 |
-
model = PeftModel.from_pretrained(base_model, "aoxo/
|
76 |
```
|
77 |
|
78 |
## Training Details
|
|
|
70 |
from peft import PeftModel, PeftConfig
|
71 |
from transformers import AutoModelForCausalLM
|
72 |
|
73 |
+
config = PeftConfig.from_pretrained("aoxo/kevin-token-compressor")
|
74 |
base_model = AutoModelForCausalLM.from_pretrained("unsloth/Meta-Llama-3.1-8B-bnb-4bit")
|
75 |
+
model = PeftModel.from_pretrained(base_model, "aoxo/kevin-token-compressor")
|
76 |
```
|
77 |
|
78 |
## Training Details
|