crumb commited on
Commit
ce748b7
·
1 Parent(s): 94d925e

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +9 -4
README.md CHANGED
@@ -23,15 +23,20 @@ model = AutoModelForCausalLM.from_pretrained(
23
  )
24
  tokenizer = AutoTokenizer.from_pretrained(model_id)
25
 
26
- input_ids = tokenizer("Once upon a time", return_tensors="pt").input_ids
27
- outputs = model.generate(input_ids, max_new_tokens=64, temperature=0.8, do_sample=True)
28
  print(tokenizer.batch_decode(outputs)[0])
29
  """
30
- <s> Once upon a time after a high-profile lawsuit, the FBI will move further into the investigation, according to Reuters
31
 
32
- The FBI will have the ability to identify and prosecute a person who, according to Reuters, is a former FBI agent who was arrested and charged with the crime of plot
 
 
 
 
33
  """
34
  ```
 
35
 
36
  ### Parameter count
37
 
 
23
  )
24
  tokenizer = AutoTokenizer.from_pretrained(model_id)
25
 
26
+ input_ids = tokenizer("In a shocking finding, scientist discovered a herd of unicorns living in a remote, previously unexplored valley, in the Andes Mountains. Even more surprising to the researchers was the fact that the unicorns spoke perfect English.", return_tensors="pt").input_ids.cuda()
27
+ outputs = model.generate(input_ids, max_new_tokens=128, temperature=0.7, do_sample=True, top_p=0.95, repetition_penalty=1.1)
28
  print(tokenizer.batch_decode(outputs)[0])
29
  """
30
+ <s> In a shocking finding, scientist discovered a herd of unicorns living in a remote, previously unexplored valley, in the Andes Mountains. Even more surprising to the researchers was the fact that the unicorns spoke perfect English. It turns out that only a few people believed their father’s name had been made public by their mother and her father.
31
 
32
+ As the study finds, scientists discovered the secretive organic matter in the Andes mountains: “In a forest surrounded by a stream of lakes, an unseen rock formed with a series of tuberous orbits.”
33
+
34
+ They found that the mysterious bodies were buried in some parts of the mountain, known as the Andes mountain. The researchers then searched for the body, which they identified with the Butterfly.
35
+
36
+ The discovery of the body is the result of
37
  """
38
  ```
39
+ (it's... a little undertrained! thats okay!)
40
 
41
  ### Parameter count
42