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.
|