aoxo commited on
Commit
3159d9f
1 Parent(s): 3ef59be

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
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/llama-token-compressor")
74
  base_model = AutoModelForCausalLM.from_pretrained("unsloth/Meta-Llama-3.1-8B-bnb-4bit")
75
- model = PeftModel.from_pretrained(base_model, "aoxo/llama-token-compressor")
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