bnjmnmarie
commited on
Commit
•
f80f1a7
1
Parent(s):
cb8a977
Update README.md
Browse files
README.md
CHANGED
@@ -54,9 +54,9 @@ model = PeftModel.from_pretrained(model, "kaitchup/Llama-2-7b-mt-Dutch-to-Englis
|
|
54 |
Then, run the model as follows:
|
55 |
|
56 |
```
|
57 |
-
|
58 |
|
59 |
-
prompt =
|
60 |
|
61 |
tokenized_input = tokenizer(prompt, return_tensors="pt")
|
62 |
input_ids = tokenized_input["input_ids"].cuda()
|
@@ -71,7 +71,7 @@ generation_output = model.generate(
|
|
71 |
)
|
72 |
for seq in generation_output.sequences:
|
73 |
output = tokenizer.decode(seq, skip_special_tokens=True)
|
74 |
-
print(output.split("###>
|
75 |
```
|
76 |
|
77 |
|
|
|
54 |
Then, run the model as follows:
|
55 |
|
56 |
```
|
57 |
+
my_text = "Het is voor jou."
|
58 |
|
59 |
+
prompt = my_text+" ###>"
|
60 |
|
61 |
tokenized_input = tokenizer(prompt, return_tensors="pt")
|
62 |
input_ids = tokenized_input["input_ids"].cuda()
|
|
|
71 |
)
|
72 |
for seq in generation_output.sequences:
|
73 |
output = tokenizer.decode(seq, skip_special_tokens=True)
|
74 |
+
print(output.split("###>")[1].strip())
|
75 |
```
|
76 |
|
77 |
|