Sandiago21 commited on
Commit
485cc60
1 Parent(s): 6b43e2c

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
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("chainyo/alpaca-lora-7b")
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,