Ajmalps commited on
Commit
443092c
·
verified ·
1 Parent(s): f17e6b9

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -19,9 +19,9 @@ Jump straight into using Boomer-4b:
19
  import torch
20
  from transformers import AutoTokenizer, AutoModelForCausalLM
21
 
22
- tokenizer = AutoTokenizer.from_pretrained("budecosystem/Boomer-4b", trust_remote_code=True)
23
  model = AutoModelForCausalLM.from_pretrained("budecosystem/Boomer-4b", torch_dtype=torch.bfloat16)
24
- inputs = tokenizer("Quantum mechanics is ", return_tensors="pt")
25
  sample = model.generate(**inputs, max_length=128)
26
  print(tokenizer.decode(sample[0]))
27
  ```
 
19
  import torch
20
  from transformers import AutoTokenizer, AutoModelForCausalLM
21
 
22
+ tokenizer = AutoTokenizer.from_pretrained("budecosystem/Boomer-4b")
23
  model = AutoModelForCausalLM.from_pretrained("budecosystem/Boomer-4b", torch_dtype=torch.bfloat16)
24
+ inputs = tokenizer("Newton's second law", return_tensors="pt")
25
  sample = model.generate(**inputs, max_length=128)
26
  print(tokenizer.decode(sample[0]))
27
  ```