PEFT
Safetensors
Generated from Trainer
khushimalik53 commited on
Commit
09b07f4
·
verified ·
1 Parent(s): 5a3ec20

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -113,8 +113,8 @@ Step Training Loss
113
 
114
  from transformers import AutoTokenizer, AutoModelForCausalLM
115
 
116
- tokenizer = AutoTokenizer.from_pretrained("path/to/your/fine-tuned-model")
117
- model = AutoModelForCausalLM.from_pretrained("path/to/your/fine-tuned-model")
118
 
119
  prompt = "### Instruction:\nExplain what this function does:\ndef reverse_string(s): return s[::-1]\n\n### Response:\n"
120
  inputs = tokenizer(prompt, return_tensors="pt").to(model.device)
 
113
 
114
  from transformers import AutoTokenizer, AutoModelForCausalLM
115
 
116
+ tokenizer = AutoTokenizer.from_pretrained("khushimalik53/codex-finetune")
117
+ model = AutoModelForCausalLM.from_pretrained("khushimalik53/codex-finetune")
118
 
119
  prompt = "### Instruction:\nExplain what this function does:\ndef reverse_string(s): return s[::-1]\n\n### Response:\n"
120
  inputs = tokenizer(prompt, return_tensors="pt").to(model.device)