NNEngine
/

qwen2-0.5b-python-lora

Model card Files Files and versions

NNEngine commited on 8 days ago

Commit

a9c333d

·

verified ·

1 Parent(s): 7229188

Update README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -94,9 +94,9 @@ from transformers import AutoTokenizer, AutoModelForCausalLM
 from peft import PeftModel
 base_model = AutoModelForCausalLM.from_pretrained("Qwen/Qwen2-0.5B")
-tokenizer = AutoTokenizer.from_pretrained("your-username/qwen2-0.5b-python-lora")
-model = PeftModel.from_pretrained(base_model, "your-username/qwen2-0.5b-python-lora")
 model.eval()
 ```

 from peft import PeftModel
 base_model = AutoModelForCausalLM.from_pretrained("Qwen/Qwen2-0.5B")
+tokenizer = AutoTokenizer.from_pretrained("NNEngine/qwen2-0.5b-python-lora")
+model = PeftModel.from_pretrained(base_model, "NNEngine/qwen2-0.5b-python-lora")
 model.eval()
 ```