MrOvkill commited on
Commit
0ffc2ad
1 Parent(s): 952e6c6

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -53,8 +53,8 @@ from transformers import AutoTokenizer, AutoModelForCausalLM
53
  import torch
54
 
55
  # Loading
56
- tokenizer = AutoTokenizer.from_pretrained("./merge/")
57
- model = AutoModelForCausalLM.from_pretrained("./merge/")
58
 
59
  # Completion function
60
  def infer(prompt, **kwargs):
 
53
  import torch
54
 
55
  # Loading
56
+ tokenizer = AutoTokenizer.from_pretrained("MrOvkill/Phi-3-Instruct-Bloated")
57
+ model = AutoModelForCausalLM.from_pretrained("MrOvkill/Phi-3-Instruct-Bloated")
58
 
59
  # Completion function
60
  def infer(prompt, **kwargs):