Update README.md
Browse files
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("
|
45 |
-
model = AutoModelForCausalLM.from_pretrained("
|
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 |
{
|