Aspik101 commited on
Commit
7d842a5
1 Parent(s): e9c656d

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -18,7 +18,7 @@ from transformers import LlamaTokenizer, LlamaForCausalLM, GenerationConfig
18
  tokenizer = LLaMATokenizer.from_pretrained("decapoda-research/llama-13b-hf")
19
 
20
  model = LLaMAForCausalLM.from_pretrained(
21
- "decapoda-research/llama-13b-hf",
22
  load_in_8bit=True,
23
  device_map="auto",
24
  )
 
18
  tokenizer = LLaMATokenizer.from_pretrained("decapoda-research/llama-13b-hf")
19
 
20
  model = LLaMAForCausalLM.from_pretrained(
21
+ "decapoda-research/llama-7b-hf",
22
  load_in_8bit=True,
23
  device_map="auto",
24
  )