diabolic6045 commited on
Commit
aecf642
·
verified ·
1 Parent(s): 15bccc2

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +4 -4
README.md CHANGED
@@ -149,11 +149,11 @@ The comparison reveals several key differences between ELN and standard training
149
  ## Usage
150
 
151
  ```python
152
- from peft import PeftModel
153
- from transformers import AutoModelForCausalLM
154
 
155
- base_model = AutoModelForCausalLM.from_pretrained("meta-llama/Llama-3.2-1B")
156
- model = PeftModel.from_pretrained(base_model, "diabolic6045/ELN-llama-1B-adapter")
157
  ```
158
 
159
  ## Framework Versions
 
149
  ## Usage
150
 
151
  ```python
152
+ # Load model directly
153
+ from transformers import AutoTokenizer, AutoModelForCausalLM
154
 
155
+ tokenizer = AutoTokenizer.from_pretrained("diabolic6045/ELN-Llama-1B")
156
+ model = AutoModelForCausalLM.from_pretrained("diabolic6045/ELN-Llama-1B")
157
  ```
158
 
159
  ## Framework Versions