Update README.md
Browse files
README.md
CHANGED
|
@@ -35,4 +35,9 @@ input_text = "A funny prompt would be "
|
|
| 35 |
input_ids = tokenizer(input_text, return_tensors="pt").to("cuda")
|
| 36 |
|
| 37 |
outputs = model.generate(**input_ids, max_new_tokens=100)
|
| 38 |
-
print(tokenizer.decode(outputs[0]))
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 35 |
input_ids = tokenizer(input_text, return_tensors="pt").to("cuda")
|
| 36 |
|
| 37 |
outputs = model.generate(**input_ids, max_new_tokens=100)
|
| 38 |
+
print(tokenizer.decode(outputs[0]))
|
| 39 |
+
```
|
| 40 |
+
|
| 41 |
+
## Notice
|
| 42 |
+
|
| 43 |
+
Zamba is a pretrained base model and therefore does not have any moderation mechanism.
|