Aspik101
/

Alpaca7b_python_assistant

Model card Files Files and versions

Aspik101 commited on Apr 2, 2023

Commit

1a55701

·

1 Parent(s): bbde085

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -19,7 +19,7 @@ from transformers import LlamaTokenizer, LlamaForCausalLM, GenerationConfig
 tokenizer = LLaMATokenizer.from_pretrained("decapoda-research/llama-13b-hf")
 model = LLaMAForCausalLM.from_pretrained(
-    "decapoda-research/llama-13b-hf",
     load_in_8bit=True,
     device_map="auto",
 )

 tokenizer = LLaMATokenizer.from_pretrained("decapoda-research/llama-13b-hf")
 model = LLaMAForCausalLM.from_pretrained(
+    "decapoda-research/llama-7b-hf",
     load_in_8bit=True,
     device_map="auto",
 )