Sandiago21
commited on
Commit
·
821b86d
1
Parent(s):
0addb44
Update README.md
Browse files
README.md
CHANGED
@@ -99,7 +99,7 @@ Use the code below to get started with the model.
|
|
99 |
import torch
|
100 |
from transformers import GenerationConfig, LlamaTokenizer, LlamaForCausalLM
|
101 |
|
102 |
-
tokenizer = LlamaTokenizer.from_pretrained("
|
103 |
model = LlamaForCausalLM.from_pretrained(
|
104 |
"Sandiago21/llama-13b-hf-prompt-answering",
|
105 |
load_in_8bit=True,
|
|
|
99 |
import torch
|
100 |
from transformers import GenerationConfig, LlamaTokenizer, LlamaForCausalLM
|
101 |
|
102 |
+
tokenizer = LlamaTokenizer.from_pretrained("Sandiago21/llama-13b-hf-prompt-answering")
|
103 |
model = LlamaForCausalLM.from_pretrained(
|
104 |
"Sandiago21/llama-13b-hf-prompt-answering",
|
105 |
load_in_8bit=True,
|