cerebras
/

btlm-3b-8k-base

Text Generation

Model card Files Files and versions Community

mabrowning commited on Jul 24, 2023

Commit

302c92b

•

1 Parent(s): a9e1351

Correct typo in usage

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -41,7 +41,7 @@ from transformers import AutoTokenizer, AutoModelForCausalLM
 # Load the tokenizer and model
 tokenizer = AutoTokenizer.from_pretrained("cerebras/btlm-3b-8k-base")
-model = AutoModelForCausalLM.from_pretrained("cerebras/-3b-8k-base", trust_remote_code=True, torch_dtype="auto")
 # Set the prompt for generating text
 prompt = "Albert Einstein was known for "

 # Load the tokenizer and model
 tokenizer = AutoTokenizer.from_pretrained("cerebras/btlm-3b-8k-base")
+model = AutoModelForCausalLM.from_pretrained("cerebras/btlm-3b-8k-base", trust_remote_code=True, torch_dtype="auto")
 # Set the prompt for generating text
 prompt = "Albert Einstein was known for "