Update README.md
Browse files
README.md
CHANGED
@@ -24,6 +24,5 @@ prompt = "Your text here."
|
|
24 |
formatted_prompt = f"### BEGIN CONVERSATION ###\n\n## Speaker_0: ##\n{prompt}\n\n## Walls1337bot: ##\n"
|
25 |
inputs = tokenizer(formatted_prompt, return_tensors="pt")
|
26 |
outputs = model.generate(input_ids=inputs["input_ids"].to("cuda"), max_new_tokens=128)
|
27 |
-
text_output =
|
28 |
print(tokenizer.batch_decode(outputs.detach().cpu().numpy(), skip_special_tokens=True)[0])
|
29 |
```
|
|
|
24 |
formatted_prompt = f"### BEGIN CONVERSATION ###\n\n## Speaker_0: ##\n{prompt}\n\n## Walls1337bot: ##\n"
|
25 |
inputs = tokenizer(formatted_prompt, return_tensors="pt")
|
26 |
outputs = model.generate(input_ids=inputs["input_ids"].to("cuda"), max_new_tokens=128)
|
|
|
27 |
print(tokenizer.batch_decode(outputs.detach().cpu().numpy(), skip_special_tokens=True)[0])
|
28 |
```
|