UncleFish commited on
Commit
466289f
1 Parent(s): 6c8221b

update readme

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -101,7 +101,7 @@ with torch.cuda.amp.autocast(dtype=torch.bfloat16):
101
  prediction = tokenizer.decode(generated_text[0], skip_special_tokens=True)
102
  print("==> prediciton: ", prediction)
103
  print("-"*120)
104
- # ==> prediciton: The dog is sitting on the beach and waving at his own
105
  ```
106
 
107
  More comprehensive examples can be found in the [notebook](demo.ipynb), where we provide a zero-shot and a few-shot example, respectively.
 
101
  prediction = tokenizer.decode(generated_text[0], skip_special_tokens=True)
102
  print("==> prediciton: ", prediction)
103
  print("-"*120)
104
+ # ==> prediciton: The dog is sitting on the beach and waving at his owner.
105
  ```
106
 
107
  More comprehensive examples can be found in the [notebook](demo.ipynb), where we provide a zero-shot and a few-shot example, respectively.