Sandiago21
commited on
Commit
•
485cc60
1
Parent(s):
6b43e2c
Update README.md
Browse files
README.md
CHANGED
@@ -101,7 +101,7 @@ Use the code below to get started with the model.
|
|
101 |
import torch
|
102 |
from transformers import GenerationConfig, LlamaTokenizer, LlamaForCausalLM
|
103 |
|
104 |
-
tokenizer = LlamaTokenizer.from_pretrained("
|
105 |
model = LlamaForCausalLM.from_pretrained(
|
106 |
"Sandiago21/llama-7b-hf-prompt-answering",
|
107 |
load_in_8bit=True,
|
|
|
101 |
import torch
|
102 |
from transformers import GenerationConfig, LlamaTokenizer, LlamaForCausalLM
|
103 |
|
104 |
+
tokenizer = LlamaTokenizer.from_pretrained("Sandiago21/llama-7b-hf-prompt-answering")
|
105 |
model = LlamaForCausalLM.from_pretrained(
|
106 |
"Sandiago21/llama-7b-hf-prompt-answering",
|
107 |
load_in_8bit=True,
|