update readme
Browse files
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:
|
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.
|