ragraph-ai commited on
Commit
644bb45
·
verified ·
1 Parent(s): 1ea2690

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -41,8 +41,8 @@ It's a finetune of https://huggingface.co/stabilityai/stable-code-instruct-3b tr
41
  import torch
42
  from transformers import AutoModelForCausalLM, AutoTokenizer
43
 
44
- tokenizer = AutoTokenizer.from_pretrained("lakkeo/stable-cypher-instruct-3b", trust_remote_code=True)
45
- model = AutoModelForCausalLM.from_pretrained("lakkeo/stable-cypher-instruct-3b", torch_dtype=torch.bfloat16, trust_remote_code=True)
46
 
47
  messages = [
48
  {
 
41
  import torch
42
  from transformers import AutoModelForCausalLM, AutoTokenizer
43
 
44
+ tokenizer = AutoTokenizer.from_pretrained("ragraph-ai/stable-cypher-instruct-3b", trust_remote_code=True)
45
+ model = AutoModelForCausalLM.from_pretrained("ragraph-ai/stable-cypher-instruct-3b", torch_dtype=torch.bfloat16, trust_remote_code=True)
46
 
47
  messages = [
48
  {